版本回退主要是为了解决新版本的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