centos 7 安装CUDA9.0 +CUDNN
2018-05-01 本文已影响0人
Persistently
先根据系统实际情况下载对应的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
-
选择合适的版本(与CUDA版本对应)
CUDNN
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*
-
yum update