1. 下載cudnn10.1:
官網(wǎng):https://developer.nvidia.com/rdp/cudnn-archive

可以下載兩種安裝文件:
我使用的是第一種安裝方法.
①按上圖下載的是cudnn-*tgz的壓縮包時(shí)
解壓下載的文件,可以看到cuda文件夾,在當(dāng)前目錄打開終端,執(zhí)行如下命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
執(zhí)行安裝,其實(shí)就是拷貝頭文件和庫文件并給予權(quán)限;
查看cudnn版本
在終端輸入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果出現(xiàn)下圖所示版本信息,說明安裝成功。

②下載另三個(gè)文件進(jìn)行安裝:

sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb
參考: