1. 查看 cuda 和 cudnn 版本
命令:
cat /usr/local/cuda/version.txt
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
2. 安裝
從官網(https://developer.nvidia.com/rdp/cudnn-download)上下載需要的安裝包:cudnn-8.0-linux-x64-v6.0.tgz
解壓處理:tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz
解壓後得到一個cuda文件夾:
進入cuda文件夾,裏面有:一個include文件夾,和一個lib64文件夾,複製一些東西到我們之前安裝cuda的路徑下(/usr/local/cuda/)
命令:
$cd cuda
$sudo cp lib64/lib* /usr/local/cuda/lib64/
$sudo cp include/cudnn.h /usr/local/cuda/include/
3. 更新cuDNN庫文件的軟連接
命令:
$ sudo chmod +r libcudnn.so.6.0.21
$ sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6
$ sudo ln -sf libcudnn.so.6 libcudnn.so
$ sudo ldconfig
4. 檢驗
命令:cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
5. 出現結果,成功