安装PyTorch后引入torchvision报错:Failed to load image Python extension: warn(f“Failed to load image Python
在安装PyTorch后,import torch没有报错,而import torchvision报错了,所以多半是因为pytorch和torchvision版本没对上,指定torchvision安装版本即可
·
一、问题分析:
报错如下:
Failed to load image Python extension: warn(f"Failed to load image Python extension: {e}")
通常出现以上报错可能有两个原因供参考:
1、安装的pytorch和torchvision版本不匹配
2、使用的cuda版本不合适
由于在安装PyTorch后,import torch没有报错,而import torchvision报错了,所以多半是因为pytorch和torchvision版本没对上。
二、解决方法:
1、查找pytorch和torchvision的版本对应关系
最新的版本对应关系查询地址:https://github.com/pytorch/vision#installation
2、指定torchvision安装版本
以下是我指定的安装参数
查表,1.12.0版本的pytorch对应0.12版本的torchvision,指定安装0.12版本的torchvision即可。
原安装命令:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
修改安装命令:
conda install pytorch torchvision==0.12 torchaudio cpuonly -c pytorch
再次测试,成功引入:
更多推荐
已为社区贡献2条内容
所有评论(0)