- 博客(16)
- 收藏
- 关注
原创 复现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 150
原创 装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 100
原创 win10 3060 cuda11.6 pytorch1.13.0 配语义分割环境
win10 3060 cuda11.6 pytorch1.13.0 mmsegmentation0.30.0搭建语义分割demo'环境
2023-08-11 22:15:05 252
原创 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 1653
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人