No module named 'torchvision.ops'的解决办法

问题: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即可
No module named 'torchvision.ops'的解决办法_第1张图片

不同版本的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

参考文献

torch版本和cuda对应的参考链接
如何解决pytorch 编译时CUDA版本与运行时CUDA版本不对应

你可能感兴趣的:(error,cuda,pytorch,torchvision.ops)