1. 程式人生 > >ubuntu18.04中安裝cuDNN

ubuntu18.04中安裝cuDNN

udnn安裝注意點:
cudnn的安裝其實很簡單,關鍵點是一定要安裝cuda對應的cudnn包,本機中安裝的cuda7.5所以對應的cudnn為v5.1這很重要,我就是安裝錯了版本,導致後面caffe的編譯總是出錯。
cudnn安裝步驟:
1、從官網上下載cudnn的安裝包。
2、將安裝包解壓,將此安裝包放在home路徑下即可,並在當前路徑下進行解壓,解壓後的資料夾名為cuda。
3、在終端上編輯如下程式碼:

$ cd cuda/include 
$ sudo cp *.h /usr/local/cuda/include/ (注意這裡是你自己的cuda的安裝地址,複製*.h檔案) 

  

4、關閉終端並重新開啟(也可以返回最開始的目錄)

$ cd cuda/lib64 
$ sudo cp libcudnn* /usr/local/cuda/lib64/ 

 



5、終端執行:

$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* 

 

轉自:https://blog.csdn.net/u010480194/article/details/54287727