Ubuntu16.04离线安装Pytorch

[版本说明]

• Ubuntu16.04
• cuda9.0
• python3.7
• pytorch-1.1.0
• torchvision-0.3.0

1. 下载离线包

进入PyTorch官网

找到自己需要的版本，我需要下载的pytorch版本是1.1.0

打开这个网址后，根据自己的版本需要，点击下载。
网址上的cu表示该版本支持GPU，1.1.0表示pytorch版本，cp37代表python3.7

2. 安装torch

（1）进入pytorch安装包的下载目录，打开终端。
（2）创建并激活自己的虚拟环境。

conda create --name wdn python=3.7 # wdn是我的虚拟环境名
source activate wdn

（3）安装pytorch

pip install torch-1.1.0-cp37-cp37m-linux_x86_64.whl
# torch-1.1.0-cp37-cp37m-linux_x86_64.whl是我下载的whl文件名

3. 安装torchvision

torchvision包是服务于pytorch深度学习框架的。安装pytorch时，torchvision独立于torch。必须同时安装了torch和torchvision，才能顺利使用pytorch。
直接再终端输入：pip install torchvision==0.3.0

注意到图片底端的pillow-7.0.0

4. 验证是否安装成功

打开终端，激活安装了PyTorch和torchvision的虚拟环境，输入python进入python环境

然后输入

import torch
import torchvision

结果报错了

原因就在于刚才第3步安装torchvision中提到的：pillow-7.0.0，安装torchvision时，默认安装的pillow版本为7.0.0，此版本的pillow已经移除了’PILLOW_VERSION’，而pillow6.1还保留着，所以只需要conda install pillow=6.1即可。

再次验证

没有报任何错误，说明安装成功。

官网
torchvision：https://pypi.org/project/torchvision/0.3.0/
pytorch：https://pytorch.org/