成功解决ModuleNotFoundError: No module named 'torchvision.ops'

音响陶醉 2022-09-19 13:50:10  65897

分类专栏：资讯

问题：cuda的版本是cuda 10.0，在使用torchvision=0.2.2，torch版本为1.4.0时候报错：
No module named 'torchvision.ops'
网上说是torchvision的版本过低，应该将其至少更新到0.3.0。于是将torchvision更新到torchvision，但此时又出现了新的错误：

ImportError: libcudart.so.9.0: cannot open shared object file:
No such file or directory
原因:torchvision和cuda版本不匹配的问题
解决办法
我疑惑的是，我的cuda版本是10.0的，为什么要报这样的错误，于是我查找了torchvision版本和cuda版本对应的情况，使用nvcc -V 查看cuda的版本，然后在此链接查找对应的pytorch版本并安装，然后直接安装torchvision即可不同版本的pytorch的安装方式：

# CUDA 9.2
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 -c pytorch

# CUDA 10.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

# CPU Only
conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly -c pytorch

网站声明：如果转载，请联系本站管理员。否则一切后果自行承担。

本文链接：https://www.xckfsq.com/news/show.html?id=2773

赞同 0

评论 0 条

成功解决ModuleNotFoundError: No module named 'torchvision.ops'

相关文章

关注我们