当前位置:   article > 正文

解决pytorch与torchvision版本不兼容问题(解决UserWarning: Failed to load image Python extension: [WinError 127]_if you don't plan on using image functionality fro

if you don't plan on using image functionality from `torchvision.io`, you

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//这里根据自身需求输入版本号

等待下载完成就解决啦。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/917950
推荐阅读
相关标签
  

闽ICP备14008679号