前言
顯卡驅動以及CUDA安裝好後(https://blog.csdn.net/avideointerfaces/article/details/100558457),就可以根據cuda的版本以及平臺類型(linux,windows,ios等)等信息來下載對應的cudnn包。
下載
1)首先進入下面的nvidia官網
https://developer.nvidia.com/cudnn
2)點擊下圖紅框所示的“Download cuDNN".如果你還沒登陸的話,會彈出一個登陸頁面要你輸入用戶名和密碼。當然還沒註冊的話,先免費註冊吧。
3)註冊並登陸後,重新點擊download cuDNN,這時出現下面的頁面,你一定要在紅框裏面打勾來遵守軟件協定。
4)打完勾後,會彈出cudnn版本list,如下所示。如果這幾個版本不符合你的要求,可以點擊下面紅框,會現實full list版本。
5)因爲我的cuda版本是9.1,而cuda9.1對應3個左右的cudnn版本,我這裏選擇一箇舊一點的cudnn7.0.5。因爲平臺是Ubuntu,所以選擇linux庫,如下圖第二個紅框所示。點擊它,會立即開始下載,如下面第三個紅框所示。
安裝
很久沒有下載cudnn庫了,在上面第三個紅框中,看到後綴名爲solitairetheme8的下載包,當時覺得很納悶是不是搞錯了,應該是後綴名爲tgz啊,是不是下載錯了。折騰很久,後來才知道下載到本地後,需要手動修改後綴名爲tgz,然後再解壓縮。命令如下所示。
mv cudnn-9.1-linux-x64-v7.solitairetheme8 cudnn-7.0.5.tgz
tar -xvzf cudnn-7.0.5.tgz
因爲下載的是庫,不是源代碼,所以不需要編譯了。只需要將解壓縮出來的so庫和頭文件拷貝到系統目錄下即可。
sudo cp cuda/include/cudnn.h /usr/include/
sudo cp cuda/lib64/libcudnn* /lib