解决pytorch与torchvision版本不兼容问题(解决UserWarning: Failed to load image Python extension: [WinError 127]

努力的码 2024-08-06 14:05:02 阅读 89

1.遇到问题

每次跑python代码的时候,如果我用到了torchvision有关的东西,就会报下面这个错误

UserWarning: Failed to load image Python extension: [WinError 127] 找不到指定的程序。 If you don t plan on using image functionality from `torchvision.io`, you can ignore this warning.

这并不会影响我程序的运行。但想着,每次都报这个,感觉不舒服,就想着把它解决了

2.解决问题

问题原因:pytorch和torchvision的版本不兼容导致

下面是pytorch和torchvision兼容的对应版本,我们应该按照下图,安装对应版本的torchvision

解决步骤:

(1)先查看本地pytoch的版本

打开anaconda prompt,进入自己创建好的pytorch虚拟环境,输入命令

conda activate pytorch

输入命令,进入python环境,依次输入下面命令

python

import torch

print(torch.__version__)

如图所示,出现了我所下载的pytorch版本,是2.3.0。

查看torchvision版本,依次输入命令

import torchvision

print(torchvision.__version__)

(2)下载相应的torchvision版本(这里我所要下载的是0.18.0版本的)

重新打开anaconda prompt窗口,进入pytorch虚拟环境,输入命令

conda install torchvision==0.18.0//这里根据自身需求输入版本号

等待下载完成就解决啦。



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。