from torchvision import transforms出现报错

一、问题描述

运行from torchvision import transforms这个语句的时候PyCharm控制台输出如下：

UserWarning: Failed to load image Python extension: [WinError 126] 找不到指定的模块。   warn(f"Failed to load image Python extension: {e}")

二、分析问题

检查自己pytorch版本和torchvision版本是否对应，经过查看自己使用conda安装的pytorch版本是1.10.2，torchvision版本是0.11.3，cuda版本是10.2，版本对应关系发现没有问题

三、解决方案

降版本试试看，发现还真的成功了

首先卸载pytorch，conda activate pytorch进入虚拟环境，然后conda uninstall pytorch

可以conda list测试一下看看列表是否卸载了，发现pytorch和torchvision都没有了

进去PyTorch官网重新安装PyTorch

这里选取pytorch版本1.9.0，cuda版本10.2（这里注意我的电脑GPU驱动程序版本461.40满足要求），torchvision版本0.10.0

conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=10.2 -c pytorch

经过尝试之后，发现控制台就不报错了，运行成功。