
pytorch
文章平均质量分 75
苍蓝儿
蓝天依旧,明眸如初
展开
-
Nvidia 显卡 Failed to initialize NVML Driver/library version
https://www.zywvvd.com/notes/system/linux/driver/driver-library-version-mismatch/driver-library-version-mismatch/本文记录错误 Failed to initialize NVML: Driver/library version mismatch 错误解决方案。问题复现$ nvidia-smi -->Failed to initialize NVML: Driver/libr原创 2022-01-28 10:42:24 · 322 阅读 · 0 评论 -
从服务器映射 jupyter 到本机应用
Jupyter Notebook是基于网页的用于交互计算的应用程序,方便易用,本文记录将远程jupyter环境映射到本机的方法。简介Jupyter Notebook是基于网页的用于交互计算的应用程序。其可被应用于全过程计算:开发、文档编写、运行代码和展示结果。——Jupyter Notebook官方介绍简而言之,Jupyter Notebook是以网页的形式打开,可以在网页页面中直接编写代码和运行代码,代码的运行结果也会直接在代码块下显示。如在编程过程中需要编写说明文档,可在同一个页面中直..原创 2021-04-08 01:31:21 · 324 阅读 · 0 评论 -
将训练好的 mmdetection 模型转为 tensorrt 模型
mmdetection 是商汤科技(2018 COCO 目标检测挑战赛冠军)和香港中文大学开源的基于Pytorch实现的深度学习目标检测工具箱,性能强大,运算效率高,配置化编程,比较容易训练、测试。但pytorch模型不易于部署,运算速度还有进一步提升的空间,当前比较有效的方法是将模型转换为行为相同的tensorrt模型,本文记录转换流程。任务思路转换mmdetection 的 pytorch模型到tensorrt模型有多种方法,本文使用 mmdetection-to-tensorrt 库作为核心.原创 2021-04-08 01:26:03 · 6014 阅读 · 5 评论 -
Pytorch dataloader 错误 “DataLoader worker (pid xxx) is killed by signal” 解决方法
在使用 pytorch dataloader 时,出现了当把num_workers 设置不为0即报错的问题,本文记录两种此类错误的解决方案。Dataloader - num_workersPytorch 中加载数据的模块Dataloader有个参数num_workers,该参数表示使用dataloader时加载数据的进程数量,可以理解为为网络搬运数据的工人数量;所以如果dataloader比较复杂,工人多的时候自然可以节省大量数据加载时间,他们可以在网络训练时同时进行数据加载工作,等网络训.原创 2020-12-04 15:28:11 · 73369 阅读 · 25 评论 -
Netron 可视化网络结构
pytorch模型可视化网络结构不是特别方便,需要借助Netron工具,本文记录可视化方法。Github 官网链接支持的网络格式Netron supports ONNX (.onnx, .pb, .pbtxt), Keras (.h5, .keras), Core ML (.mlmodel), Caffe (.caffemodel, .prototxt), Caffe2 (predict_net.pb), Darknet (.cfg), MXNet (.model, -symbol.json),.原创 2020-11-20 22:25:40 · 1350 阅读 · 0 评论