安装PyTorch后引入torchvision报错:Failed to load image Python extension: warn(f“Failed to load image Python

一、问题分析:

报错如下:

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

安装PyTorch后引入torchvision报错:Failed to load image Python extension: warn(f“Failed to load image Python

2、指定torchvision安装版本

以下是我指定的安装参数

安装PyTorch后引入torchvision报错:Failed to load image Python extension: warn(f“Failed to load image Python

查表,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

再次测试,成功引入:

安装PyTorch后引入torchvision报错:Failed to load image Python extension: warn(f“Failed to load image Python

Original: https://blog.csdn.net/qq_45397526/article/details/125834669
Author: 不会写代码的小周
Title: 安装PyTorch后引入torchvision报错:Failed to load image Python extension: warn(f“Failed to load image Python

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/669971/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球