網頁

2019年9月21日 星期六

ubuntu 18.04 install cuDNN

NVIDIACUDA®深度神經網路函式庫(cuDNN)是GPU加速的用於深度神經網絡的原語庫。 cuDNN在執行上提供了高度優化的能力,像是在forward和backward卷積,池化,normalization和activation layers。以下是安裝流程:



1. 先到cudnn的document
https://docs.nvidia.com/deeplearning/sdk/cudnn-install/

2. 到NVIDIA cuDNN登入會員後下載deb檔
https://developer.nvidia.com/cudnn



2.3. Installing cuDNN on Linux會說有兩種安裝方式,選擇2.3.2. Installing from a Debian File會比較容易



安裝好三個deb檔之後,會在/usr/src新增cudnn_samples_v7資料夾

3. 驗證是否安裝完成
$ cp -r /usr/src/cudnn_samples_v7/ $HOME
$ cd  $HOME/cudnn_samples_v7/mnistCUDNN
$ make clean && make
$ ./mnistCUDNN


最後顯示Test passed!代表安裝完成



參考
https://www.twblogs.net/a/5d7ef5a1bd9eee5327ffe84d







沒有留言:

張貼留言