本人在工作时遇到了一个比较尴尬的问题,就是虚拟环境中已有固定版本的torch,但没有安装torchvision,因此很本能的安装:
pip install torchvision
如果直接这样安装的话,我们发现会连带安装新版本的torch,由于不是本人的虚拟环境,不想改变别人的torch版本。因此该方法行不通。
就在思考如何分开安装torch 和 torchvision,后来发现,只需下载较低版本的torchvision就可以解决我的问题。只要下载的版本大于0.3.x,就会下载更新torch。因此选择安装低版本的。
本人尝试了0.1.6版本(pip install torchvision == 0.1.6),发现版本过低,在使用torchvision.transforms时报错:
module 'torchvision.transforms' has no attribute 'xxxx'。
这时我们要适当调高版本,因此本人下载了0.2.1(pip install torchvision == 0.2.1),成功解决问题!
大家如果有什么问题可以继续调试版本,查阅torch和torchvision的版本对应关系,二者版本不匹配有时也会报错!