ubuntu16.04安裝cuDNN


cudnn的安裝非常簡單

(1)下載安裝文件

按需求下載cudnn的安裝文件https://developer.nvidia.com/rdp/cudnn-archive

Tar File的下載如下圖所示,選擇紅方框中的選項進行下載

 

下載的是cudnn-*tgz的壓縮包時,按下方指令進行安裝:

首先解壓縮下的cudnn壓縮包文件

tar -xzvf cudnn-9.0-linux-x64-v7.tgz

執行安裝,其實就是拷貝頭文件和庫文件並給予權限

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

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

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

(2)驗證安裝是否成功

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果顯示

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 2
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

則cudnn安裝成功


免責聲明!

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



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