问题描述
import torch没有问题,正常。但是import torchvision出错
查找相关资料说是torch和torchvision的版本不匹配。但是在conda list下发现版本是完全匹配的!
怀疑是玄学出错所以索性重装了。conda uninstall torchvision了一下然后conda install torchvision。但是问题并没有解决
发现是由于python解释器默认的是base环境下的!在我的base环境下只有一个torch而没有torchvision。而我的torch和torchvision是安装在虚拟的子环境(叫pytorch)下的,所以pycharm要换到子环境的python解释器下。
解决方案
在右下角切换python解释器,切换到子环境(pytorch)下。
然后再run就能成功了 。