程序员最近都爱上了这个网站  程序员们快来瞅瞅吧!  it98k网:it98k.com

本站消息

站长简介/公众号

  出租广告位,需要合作请联系站长

+关注
已关注

分类  

暂无分类

标签  

暂无标签

日期归档  

2021-12(24)

2022-01(13)

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

发布于2022-10-05 08:44     阅读(1140)     评论(0)     点赞(21)     收藏(1)


一、问题分析:

报错如下:

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

再次测试,成功引入:



所属网站分类: 技术文章 > 博客

作者:病毒快消失

链接:https://www.pythonheidong.com/blog/article/1786129/09bcb0acb5805cb1e9b6/

来源:python黑洞网

任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任

21 0
收藏该文
已收藏

评论内容:(最多支持255个字符)