Ubuntu 16.04.7使用python 2.7.12和pip 20.3.4安装pytorch指南

Ubuntu 16.04.7使用python 2.7.12安装pytorch指南

    • pytorch的安装与版本问题
    • 版本对应表格
    • 开始安装
      • 请确认你的python、pip版本:
      • 请查看你的显卡信息和确认你的CUDA版本
      • 确认你需要安装的pytorch和torchvision版本
      • 测试pytorch

pytorch的安装与版本问题

因为需要在一些旧版工程上进行工作,特意新装了ubuntu16.04.7并使用其自带的python 2.7.12。

根据网上的资料显示,pytorch在2020年就已经停止了对python 2.7的支持,新版本的pytorch已经无法在python 2.7环境中部署。部署旧版本pytorch变得困难起来,因为pytorch官网提供的pip指令安装较新版本(>=1.5.0)仅针对python 3.X环境,因此需要选择合适的旧版本pytorch才能通过pip指令安装。

根据我的N次试错,python 2.7.12所能安装的最新版本是1.5.0+cu92
但不推荐采用此版本,因为按照pytorch官方的说法,1.5.0和更高版本是不能在python2.7部署的,torchvision官方版本表格中也显示的很清楚。但我确实部署成功了,这就有点尴尬了。不过当我尝试安装torchvision时,会自动重装1.4.0版本,因此1.5.0确实还是在python2.7的支持上存在问题。

你可以通过如下的指令安装pytorch:

// 仅安装pytorch
pip install torch==X.X.X+cuXX -f https://download.pytorch.org/whl/torch_stable.html

版本对应表格

源自torchvision的GitHub:

torch torchvision python
1.8.0 0.9.0 >=3.6
1.7.1 0.8.2 >=3.6
1.7.0 0.8.1 >=3.6
1.7.0 0.8.0 >=3.6
1.6.0 0.7.0 >=3.6
1.5.1 0.6.1 >=3.5
1.5.0 0.6.0 >=3.5
1.4.0 0.5.0 ==2.7,>=3.5,<=3.8
1.3.1 0.4.2 ==2.7,>=3.5,<=3.7
1.3.0 0.4.1 ==2.7,>=3.5,<=3.7
1.2.0 0.4.0 ==2.7,>=3.5,<=3.7
1.1.0 0.3.0 ==2.7,>=3.5,<=3.7
<=1.0.1 0.2.2 ==2.7,>=3.5,<=3.7

开始安装

请确认你的python、pip版本:

python
pip -V

我使用的是python 2.7.12 和pip 20.3.4
注意:此版本的pip需要使用pip官方提供的get-pip.py进行手动安装。Ubuntu 16.04.7自带的pip是pip 8,它已经太过老旧以至于不能安装pytorch。请使用apt-get将其卸载后,重新手动安装新版的pip

再次注意:请千万!一定!不要直接在终端使用pip的自动升级指令!它会导致pip 21被安装到你的PC,并且它会和python 2.7产生严重冲突!解决方案只有全盘搜索pip并且逐一删除它们。请一定使用get-pip.py脚本手动安装pip

请查看你的显卡信息和确认你的CUDA版本

nvidia-smi
nvcc -V

我使用的是460.32.03版显卡驱动和CUDA 11.0
显卡驱动和CUDA版本具有向下兼容的特性(虽然有时不太可靠),采用新的驱动和CUDA版本通常情况并不会影响旧版本pytorch的部署(pytorch的兼容性还是比TF强一点的)

确认你需要安装的pytorch和torchvision版本

小技巧:你可以故意输入一个高版本的安装指令,来获得所有可安装的版本目录,然后选择合适的版本

bie@bie:~$ pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

DEPRECATION: Python 2.7 reached the end of its life on January 1st, 2020. 
Please upgrade your Python as Python 2.7 is no longer maintained. pip 21.0 will drop support for Python 2.7 in January 2021. 
More details about Python 2 support in pip can be found at https://pip.pypa.io

Defaulting to user installation because normal site-packages is not writeable

Looking in links: https://download.pytorch.org/whl/torch_stable.html

ERROR: Could not find a version that satisfies the requirement torch==1.7.0+cu110 (from versions: 1.4.0, 1.4.0+cpu, 1.4.0+cu100, 1.4.0+cu92, 1.5.0+cpu, 1.5.0+cu92)

ERROR: No matching distribution found for torch==1.7.0+cu110

我选择安装pytorch 1.4.0和torchvision 0.5.0
注意:请记得在版本号后添加合适的cuXX,以确保你安装的版本是你需要的CUDA版本,而不是pytorch默认的
安装指令:

pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

等待安装完成即可
你会看到终端显示:

Successfully installed pillow-6.2.2 torch-1.4.0+cu92 torchvision-0.5.0+cu92

至此,安装完成,可以进行测试了

测试pytorch

python
import torch
torch.__version__

输出:

'1.4.0+cu92'

如果遇到如下报错:
ImportError:No module named builtins
请手动安装future

pip install future

你可能感兴趣的:(pytorch,python,ubuntu,linux)