linux下pytorch安装(cuda9.0+cudnn7.0)

前言：在服务器上安装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

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

这样就可以了

来源：CSDN

作者：xm_ai

链接：https://blog.csdn.net/qq_41612527/article/details/104117980