PyTorch 1.0.0安装（Linux)

目录

起：

查看服务器的CUDA版本 和 显存信息

PyTorch 1.0.0 安装

创建一个pytorch环境：torch1

进入torch1环境，安装 pytorch 1.0.0

进入PyCharm 配置 PyTorch 1.0.0

安装 matplotlib

检查：所有环境已经配置好

其他问题：ImportError: cannot import name ‘PILLOW_VERSION’

运行成功

---

起：

参加华为云开发者AI青年班黑客松大赛时，ModelArts的GPU资源马上耗尽，因此，选择远程用院里的台式服务器进行模型训练。

之前一直使用tensorflow，在此次青年班中，学习过程接触学习了pytorch，感觉很有用，就准备在台式服务器安装pytorch训练。

查看服务器的CUDA版本 和 显存信息

$ cat /proc/version

$ cat /etc/issue

$ cat /proc/driver/nvidia/version

如下图，下面第1个图倒数第二行是384.145，对应第2个图的CUDA 8.0（8.0.61.GA2）

显存信息查看：

$ nvidia-smi

PyTorch 1.0.0 安装

PyTorch1.0.0安装命令查询。

对应第三个：conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch

创建一个pytorch环境：torch1

$ conda create -n torch1 python=3.6.8

进入torch1环境，安装 pytorch 1.0.0

$ source activate torch1
$ conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch

进入PyCharm 配置 PyTorch 1.0.0

pycharm版本信息如下图。

进入File > Settings > Project Interpreter，选择 Existing environment，路径选择自己的anaconda路径下envs中刚刚创建的新环境torch1，选择对应bin目录的python文件确定即可。

安装 matplotlib

检查：所有环境已经配置好

import相应包后，没有红色下划线提示错误，就表明已经安装配置好了。

其他问题：ImportError: cannot import name ‘PILLOW_VERSION’

参考文章 https://blog.csdn.net/YANGTAOxka/article/details/105251629，点击进入functional.py，改动如下：注释的是原来的语句，下面一句是改动的。

运行成功