pytorch版本回退解决No Module named 'torch.legacy'

版本回退主要是为了解决新版本的pytorch往往会舍弃一些过去会用到的类与接口,比如torch.legacy等在torch1.0.1版本中是被舍弃了,无法正常使用的,而在0.4.1中是仍然保存的。


一、Conda降级pytorch版本或安装指定版本pytorch

如果你是使用conda包管理,你可以很容易实现版本降级,你只需要指定版本即可:

# 比如你想降级到以前的v0.4.1版本
conda install pytorch=0.4.1 -c soumith 

二、PIP降级或安装指定版本pytorch

很遗憾官网并没有提供pip降级pytorch的方案,但是我们并不是没有方法,你可以参考官网的配置选中你的配置,比如python版本,CUDA版本等等,然后复制代码,将版本号改为你想要的版本,如torch-1.0.1改为torch-0.4.0,不过有些配置可能没有,因为pytorch提供的支持越来越多。

pip install http://download.pytorch.org/whl/torch-0.4.1.post1-cp27-none-macosx_10_7_x86_64.whl 
# 比如安装0.2.0版本
# pip install http://download.pytorch.org/whl/torch-0.2.0.post1-cp35-cp35m-macosx_10_7_x86_64.whl 
pip install torchvision 

三、源码安装指定版本 

用源码编译安装,直接在github寻找指定版本,安装即可 。

https://github.com/pytorch/pytorch#from-source

参考:https://ptorch.com/news/198.html

你可能感兴趣的:(Ubuntu,No,Module,named,'torch.legacy',pytorch回退,pytorch0.4.1)