一、问题描述
运行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
经过尝试之后,发现控制台就不报错了,运行成功。