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

最新推荐文章于 2024-07-05 10:07:55 发布

置顶还是少年呀

最新推荐文章于 2024-07-05 10:07:55 发布

阅读量2.1w

点赞数 4

分类专栏： error 文章标签： cuda pytorch torchvision.ops

本文链接：https://blog.csdn.net/Mark_2018/article/details/105793147

版权

error 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

问题：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