Linux下安装pytorch

大家好。本以为Linux下安装pytorch很容易，结果也是折腾了一下，我记录一下问题，供后来者免采坑，好好学习，天天向上。其实网上都有答案，但是由于网上信息量巨大，有用的信息往往石沉大海，需要花时间甄别。

依据官网给出的安装

我们进入PyTorch官网： 显示如下：

1. 红色的框 ，这里可以点开安装你需要的版本。这里要注意版本匹配的问题。
2. 绿色的框 ，后面如果使用国内高校的镜像进行安装的话，一定记得要去掉。后面会详细讲解。

版本匹配

pytorch、cuda、python的版本对齐问题。我查到的资料不是很多，大家选择版本之前需要查看自己的python版本，cuda版本，选择合适的安装方式才能work。

利用镜像安装

这里非常感谢叫我西瓜超人，解决了下载超级慢的问题，慢到忍无可忍的境界。他讲的很详细，需要的参阅。pytorch最近已经更新到了稳定版本的1.0.1，从Pytorch官网上可以看到有多种安装方式：

比较常见的方法是通过pip和conda安装，当使用conda安装的时候，可能会遇到下载过慢的问题，下载的时候可能会遇到无尽的等待。这里推荐用清华源替代默认conda源的方法，可以解决下载过慢的问题。

清华conda源地址
添加清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

另外为了保险起见，建议同时添加第三方conda源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

根据Python和CUDA选择对应的版本，然后官方给出提示可通过运行：
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
注意，一定要去掉-c pytorch，安装的时候才会默认从清华源下载相应的包，否则下载安装包还是会很慢，因此这里用命令行:

conda install pytorch torchvision cudatoolkit=9.0 

检测是否安装成功

为了确保pytorch安装成功，可以运行以下代码进行验证。

import python
import torch
import torchvision
print(torch.__version__)
print(torchvision.__version__)

我在import torchvision时遇到了如下的问题，借鉴其他博主网页，成功解决该问题：ImportError: cannot import name ‘PILLOW_VERSION’

解决方案：点开查看详情
大家遇到的问题也许各不相同，但是我们的心愿也许都大同小异，祝成功。