解决torchvision与cuda版本不匹配的问题
首先是训练yolov5模型时遇到了找不到指定模块的问题
然后查阅了许多资料，发现可能是torch、torchvision和cuda版本不匹配。原先的版本如下：
下载对应的版本地址：https://download.pytorch.org/whl/torch_stable.html
我保持原来的torch和torchvision的版本，更新了更高版本的cuda。10.2->11.3（这里的cp37对应python的版本3.7）
下载后重新安装（找到whl文件所在目录，进入虚拟环境安装）
成功安装，没报错
总结：安装虚拟环境时特别要注意版本匹配的问题~
本图文内容来源于网友网络收集整理提供，作为学习参考使用，版权属于原作者。
THE END
二维码