TensorFlow

[TensorFlow]3.安裝cuDNN v7.0.5 Runtime Library for Ubuntu16.04

安裝之前你必須先註冊或登入,註冊填寫帳號認證email後

在按一次下載頁面,點選Download cuDNN v7.0.5 (Dec 5, 2017), for CUDA 8.0

裡面有一個 cuDNN v7.0.5 Runtime Library for Ubuntu16.04 (Deb)

一個 cuDNN v7.0.5 Developer Library for Ubuntu16.04 (Deb)

一個cuDNN v7.0.5 Code Samples and User Guide for Ubuntu16.04 (Deb)

三個都把它下載回來。

目前我安裝時最新版本就是這裡,你可以看你cuda裝的版本(但是我前面文章範例都是8.0的)所以你如果照我的方法,你就要選CUDA 8.0版的但是cuDNN可以換新版(不過也不是有新就直上,也要看你之後tensorflow安裝的版本有沒有支援到這麼新的cudnn)

下載完deb檔後就來安裝


sudo dpkg -i libcudnn7_7.0.5.15-1+cuda8.0_amd64.deb

sudo dpkg -i libcudnn7-dev_7.0.5.15-1+cuda8.0_amd64.deb

sudo dpkg -i libcudnn7-doc_7.0.5.15-1+cuda8.0_amd64.deb

上面都安裝完(且都安裝很快,輸出的內容也很少),看起來沒有錯的話

就可以來驗證cuDNN你是不是有正確安裝好

 

1. 將某一個 cuDNN 樣例複製到某一可讀寫文件目錄下
$ cp -r /usr/src/cudnn_samples_v7/ $HOME
2. 進入該文件目錄, 本文選擇的是mnistCUDNN測試
$ cd $HOME/cudnn_samples_v7/mnistCUDNN
3. 編譯 mnistCUDNN 樣例.
$ make clean && make
4. 運行 mnistCUDNN 樣例.
$ ./mnistCUDNN
5. 如果安裝成功,程序的最後則會顯示
Test passed!

 

後記:恩  後來我再裝tensorflow時發現死掉了,出現了ˊ像是下列的錯誤

原因就是NVIDIA的CUDNN有新版了,但是tensorflow跟不上他的版本,所以

我就回去重新下載Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0 這版本來安裝了

一樣三個檔案都要下載 然後跟上面的步驟再走一次

 sudo dpkg -i libcudnn6_6.0.21-1+cuda8.0_amd64.deb 
sudo dpkg -i libcudnn6-dev_6.0.21-1+cuda8.0_amd64.deb 
sudo dpkg -i libcudnn6-doc_6.0.21-1+cuda8.0_amd64.deb 

裝完之後再回去tensorflow測試看看應該就會過關了。

Be the First to comment.

Leave a Comment

你的電子郵件位址並不會被公開。

(若看不到驗證碼,請重新整理網頁。)