Ubuntu中安装GPU版本pytorch

在ubuntu中安装pytorch遇到了很多问题，因为实验室服务器的cuda版本是8.0，所以需要注意版本兼容的问题。

首先我尝试了从官网下载安装包的方法，详情可见这篇博主的博文的第二部分Ubuntu16.04环境下PyTorch简易安装教程
但是我尝试之后发现

pip install torchvision

这条语句会重新安装torch，覆盖原来的torch版本，导致出现不兼容问题，使得torch.cuda.is_available()返回False。

所以我去pytorch官网
利用图中第三条语句安装好了pytorch，这一次终于torch.cuda.is_available()返回了True。
但是新的问题又出现了…
在import torchvision时出现了cannot import name’PILLOW_VERSION’的错误。
解决方法是由于pillow版本过高，利用下面语句即可解决问题。

conda install pillow=6.1

最后一个问题是：我们都知道在安装指定包的时候要在指定的python环境中，管理python需要anaconda。但我在切换anaconda环境时出现问题，无法进入指定环境。
解决方法是：先进入base环境，再退出，即可解决上述问题。

soruce activate

conda deactivate

提到anaconda管理环境，推荐一篇很好的博文Ubuntu安装anaconda 介绍、安装、配置