centos 7 安装CUDA9.0 +CUDNN

搬运一篇安装指南，备忘

CUDA

先根据系统实际情况下载对应的CUDA，这里我下载了CUDA 9.0 版本

1.检查是否安装了GPU：

   lspci | grep -i nvidia  
复制代码

2.安装gcc、g++编译器

   sudo yum install gcc    
   sudo yum install gcc-c++  
复制代码

3.安装kernel-devel

 sudo yum install kernel-devel  
复制代码

4.安装Driver，Toolkit和Samples

  sudo sh cuda_9.0.176_384.81_linux-run.run
复制代码

在这里我们是自己安装了匹配的驱动，所以第一项Driver出来的时候选择N 后面全是Y ,即可

5.在用户根目录下的.bashrc文件中加入如下内容：

export PATH=$PATH:/usr/local/cuda-9.0/bin    
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64  
保存退出
source .bashrc （立即生效文件）
如果是全环境变量（所用用户都能使用），需要在/etc/profile文件加入上面的几句话
复制代码

CUDNN

1.选择合适的版本(与CUDA版本对应)

2.解压CUDNN

tar -xzvf cudnn-9.0-linux-x64-v7.1.tgz
复制代码

复制相关文件到cuda特定目录下(我的cuda安装目录为/usr/local/cuda-9.0/)

sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64
复制代码

修改文件权限

sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda-9.0/lib64/libcudnn*
复制代码

Reference

• www.jianshu.com/p/a201b91b3…