Ubuntu中安装GPU版本pytorch

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

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

pip install torchvision

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

所以我去pytorch官网
Ubuntu中安装GPU版本pytorch_第1张图片利用图中第三条语句安装好了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 介绍、安装、配置

你可能感兴趣的:(pytorch)