NVIDIA CuDNN 安裝說明


CuDNN是專門針對Deep Learning框架設計的一套GPU計算加速方案,目前支持的DL庫包括Caffe,ConvNet, Torch7等。

CuDNN可以在官網免費獲得,注冊帳號后即可下載。官網沒有找到安裝說明,下載得到的壓縮包內也沒有Readme. 不過google一下就會找到許多說明。基本原理是把lib文件加入到系統能找到的lib文件夾里, 把頭文件加到系統能找到的include文件夾里就可以。這里把他們加到CUDA的文件夾下(參考這里

tar -xzvf cudnn-6.5-linux-R1.tgz
cd cudnn-6.5-linux-R1
sudo cp lib* /usr/local/cuda/lib64/
sudo cp cudnn.h /usr/local/cuda/include/

執行后發現還是找不到庫, 報錯

error while loading shared libraries: libcudnn.so.6.5: cannot open shared object file: No such file or directory

而lib文件夾是在系統路徑里的,用ls -al發現是文件權限的問題,因此用下述命令先刪除軟連接

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.6.5

然后修改文件權限,並創建新的軟連接

sudo chmod u=rwx,g=rx,o=rx libcudnn.so.6.5.18 
sudo ln -s libcudnn.so.6.5.18 libcudnn.so.6.5
sudo ln -s libcudnn.so.6.5 libcudnn.so

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM