Ubuntu 安裝 cuDNN(附測試)

爲深度學習所用,博主預想在Ubuntu16.04上安裝 顯卡驅動 + CUDA + cuDNN + Tensorflow-gpu + Keras + PyCharm,參考了衆多資料,最終成功將所有軟件安裝完畢,且能成功運行使用。該篇博客介紹了cuDNN的安裝教程,亦可通過下方鏈接查看其他步驟的安裝教程。

前期說明及版本對照

換源及安裝顯卡驅動

安裝 CUDA(附測試)

安裝 cuDNN(附測試)

安裝 Tensorflow-gpu 與 Keras(附測試)

設置 PyCharm

安裝教程完整版

安裝 cuDNN

可在cuDNN的官網上下載對應版本的cuDNN安裝包,官網網址:cuDNN Archive

1、安裝cuDNN

cuDNN的安裝方法有兩種,需要下載不同的安裝包:

(1).tgz 文件安裝

該方法僅需下載一個安裝包,如圖所示
在這裏插入圖片描述
下載完畢後,進入文件所在位置,解壓安裝包(其中 xxx 爲安裝包的補充名稱)

tar -xzvf cudnn-xxx.tgz

而後將cuDNN拷貝,放入CUDA對應的文件夾中,並賦予其權限

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).deb 文件安裝

該方法需要下載三個安裝包,如下圖所示
在這裏插入圖片描述
分別對這三個安裝包執行以下命令進行安裝,文件名請自行修改,要注意,必須按以下順序進行安裝,否則會因依賴問題而報錯

sudo dpkg -i libcudnn7_7.6.4.38-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.4.38-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.4.38-1+cuda9.0_amd64.deb

2、測試cuDNN

選擇 .deb 文件安裝cuDNN,默認在 “Home” 位置,會安裝有cuDNN的測試案例

在這裏插入圖片描述
進入 samples 文件中的 mnistCUDNN 文件夾下,鼠標右鍵打開終端

在這裏插入圖片描述
編譯文件

make clean && make

編譯後執行

./mnistCUDNN

若最後結果顯示 Test passed,則說明測試成功

在這裏插入圖片描述

發佈了8 篇原創文章 · 獲贊 3 · 訪問量 1487
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章