torchvision报错：UserWarning: Failed to load image Python extension: Could not find module

torchvision报错：UserWarning: Failed to load image Python extension: Could not find module

原因

可能是torch和torchvision的版本不匹配，你可以查看一下对应网址在

https://github.com/pytorch/vision#installation

torch	torchvision	python
main / nightly	main / nightly	>=3.7, <=3.9
1.10.2	0.11.3	>=3.6, <=3.9
1.10.1	0.11.2	>=3.6, <=3.9
1.10.0	0.11.1	>=3.6, <=3.9
1.9.1	0.10.1	>=3.6, <=3.9
1.9.0	0.10.0	>=3.6, <=3.9
1.8.2	0.9.2	>=3.6, <=3.9
1.8.1	0.9.1	>=3.6, <=3.9
1.8.0	0.9.0	>=3.6, <=3.9
1.7.1	0.8.2	>=3.6, <=3.9
1.7.0	0.8.1	>=3.6, <=3.8
1.7.0	0.8.0	>=3.6, <=3.8
1.6.0	0.7.0	>=3.6, <=3.8
1.5.1	0.6.1	>=3.5, <=3.8
1.5.0	0.6.0	>=3.5, <=3.8
1.4.0	0.5.0	==2.7, >=3.5, <=3.8
1.3.1	0.4.2	==2.7, >=3.5, <=3.7
1.3.0	0.4.1	==2.7, >=3.5, <=3.7
1.2.0	0.4.0	==2.7, >=3.5, <=3.7
1.1.0	0.3.0	==2.7, >=3.5, <=3.7
<=1.0.1	0.2.2	==2.7, >=3.5, <=3.7

如果你忘记了你的torch与torchvision的版本，你可以在Anaconda Prompt中按照：

# ltorch是我的环境名
conda activate ltorch 
python
import torch
import torchvision
torch.__version__
torchvision.__version__

下图是我的torch和torchvision的版本

解决

比如我的torch版本是1.9.1，cuda版本是10.2，按照表格对应的torchvision版本是0.10.1，那么就可以在Anaconda Prompt中输入：

conda activate ltorch # ltorch是我建立的虚拟环境名
pip install torchvision==0.10.1+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

重新import，不报错。

感谢

感谢@qq_43326730指出原文的错误。