- 博客(16)
- 收藏
- 关注
RSS订阅原创 复现swav
3.RuntimeError: Distributed package doesn‘t have NCCL built in，因为window不支持nccl后端通信。下载链接：https://github.com/facebookresearch/swav。大功告成，运行完记得把checkpoint.pth移开，才能再次运行，不然有报错。方法：在上面加一句os.environ[“RANK”] = ‘0’然后她是默认是多gpu运行的，会有几个报错。
2023-09-19 20:18:35
109
15原创 装cuda和torch时的问题记录
机缘巧合下发现原来torch11.3版本是自带cuda11.6的，我在anaconda中另外create一个虚拟环境torch101，再用nvcc -V查看cuda版本就是系统的cuda版本11.5。一开始我在anaconda的base环境中用nvcc -V查看cuda版本是11.5与电脑的版本号符合，但是在虚拟环境pytorch中查的粗大版本是11.6。怎么也弄不明白，网上也没有。
2023-08-12 20:33:14
89
原创 win10 3060 cuda11.6 pytorch1.13.0 配语义分割环境
win10 3060 cuda11.6 pytorch1.13.0 mmsegmentation0.30.0搭建语义分割demo'环境
2023-08-11 22:15:05
179
原创 yolov5的.pt模型转.onnx模型并运行
运行export把pt转成onnx之前先做一下修改：data是训练.pt文件时的yaml文件，weights是需要转成onnx模型的pt模型的路径，batch-size改成1，想用cpu运行device就改成cpu，想用gpu运行就改成0，include就写onnx。直接在detect.py运行onnx模型：weights中放入onnx模型路径，如果想检测图片就在source放入图片路径，要检测摄像头就设置为0，data是训练.pt文件时的yaml文件。
2023-05-29 10:34:58
1438
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人