一、安裝cuDNN
在CentOS上,NVIDIA驅動的RPM包依賴於一些外在的包(external packages)
# 從EPEL上面來安裝dkms
yum install epel-release
yum install dkms
註冊賬號,下載cuDNN
網址:https://developer.nvidia.com/rdp/cudnn-download
# 解壓solitairetheme8文件
tar -zxvf cudnn-9.0-linux-x64-v7.1.solitairetheme8
將下述文件拷貝到CUDA Toolkit文件夾,並更改權限
cp cuda/include/cudnn.h /usr/local/cuda/include
cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*