先根據(jù)系統(tǒng)實際情況下載對應(yīng)的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
在這里我們是自己安裝了匹配的驅(qū)動界睁，所以第一項Driver出來的時候選擇N 后面全是Y ,即可
5.在用戶根目錄下的.bashrc文件中加入如下內(nèi)容：
export PATH=$PATH:/usr/local/cuda-9.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
保存退出
source .bashrc （立即生效文件）
如果是全環(huán)境變量（所用用戶都能使用）球昨，需要在/etc/profile文件加入上面的幾句話
CUDNN
-
選擇合適的版本(與CUDA版本對應(yīng))
CUDNN
2.解壓CUDNN
tar -xzvf cudnn-9.0-linux-x64-v7.1.tgz -
復(fù)制相關(guān)文件到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 -
修改文件權(quán)限
sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda-9.0/lib64/libcudnn*
-
yum update
