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

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

等待下载完成就解决啦。