centos7安装CUDA9.0+CUDNN

搬运一篇安装指南&＃xff0c;备忘

CUDA

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

1.检查是否安装了GPU&＃xff1a;

lspci | grep -i nvidia


2.安装gcc、g&＃43;&＃43;编译器

sudo yum install gcc sudo yum install gcc-c&＃43;&＃43;


3.安装kernel-devel

sudo yum install kernel-devel


4.安装Driver&＃xff0c;Toolkit和Samples

sudo sh cuda_9.0.176_384.81_linux-run.run


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

5.在用户根目录下的.bashrc文件中加入如下内容&＃xff1a;

export PATH&＃61;$PATH:/usr/local/cuda-9.0/bin
export LD_LIBRARY_PATH&＃61;$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
保存退出
source .bashrc &＃xff08;立即生效文件&＃xff09;
如果是全环境变量&＃xff08;所用用户都能使用&＃xff09;&＃xff0c;需要在/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&＃43;r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda-9.0/lib64/libcudnn*


Reference

• www.jianshu.com/p/a201b91b3…