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指出原文的错误。