NVIDIA CuDNN 安裝說明

2020-06-16 18:02:08

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

本文永久更新連結地址：http://www.linuxidc.com/Linux/2015-04/116445.htm

NVIDIA CuDNN 安裝說明

熱門文章