前言:在服务器上安装pytorch
1、检查驱动+CUDA+CUDNN
1、在安装pytorch之前,我知道我的服务器上已经安装了驱动,查看驱动命令
nvidia-smi
如果是这样,证明已经安装了驱动
2、检查安装CUDA命令
nvcc -V
可以看到结果CUDA版本9.0
3、查看安装CUDNN命令
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
可以看到结果CUDNN版本7.0
2、安装pytorch
1、创建一个环境变量(已经安装好了anaconda)
conda create -n pytorch pip python=3.7.1
2、激活环境变量
source activate pytorch
3、检查自己操作系统版本
cat /etc/lsb-release
操作系统 Ubuntu
版本号16.04
4、点击这里,进入pytorch配置网站
这里注意,我之前使用的是CUDA9.0版本,这里是9.2 版本不符,所以需要有所改动,请看下面
5、由于这种方法安装pytorch很慢,所以我采用了清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
检查CUDA版本命令:
cat /usr/local/cuda/version.txt
6、由于我的CUDA版本本来就是9.0,所以安装pytorch对应CUDA版本就是9.0
conda install pytorch torchvision cudatoolkit=9.0
7、最后我的结果是这样的
测试安装成功
这里需要说明的是在我python后输入
import torchvision
会报错如下,这是因为pillow7.0.0已经没有PILLOW_VERSION这个东西了,而pillow6.1还保留着
所以只需要
conda install pillow=6.1
这样就可以了