目录
其他问题: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 安装
对应第三个: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,改动如下:注释的是原来的语句,下面一句是改动的。