Ubuntu 18.04 安裝 Tensorflow 環境之三:安裝 CUDNN 7.4.1

 

CUDA 10 就安裝好了,下來需要安裝 cudnn 7.4.1,才能利用 GPU 進行計算加速。

安裝過程

1、下載CUDNN

NVIDIA官網上下載 NVIDIA CUDNN 相關資源,NVIDIA 官網需要註冊,在 login 進去之後,點擊同意“I Agree To the Terms of the cuDNN Software License Agreement”,相關資源纔會顯示。

在官網上下載4個文件,如果不想進行註冊,或需要翻牆才能註冊,或由於網速差等不能下載的,我已上傳相關資源,可點擊每個資源後面的鏈接,進行下載。

要下載四個文件
cuDNN Library for Linux  國內下載地址
cuDNN Runtime Library for Ubuntu18.04 (Deb)  國內下載地址
cuDNN Developer Library for Ubuntu18.04 (Deb)  國內下載地址
cuDNN Code Samples and User Guide for Ubuntu18.04 (Deb)  國內下載地址

2、將下載好的文件,傳到Ubuntu服務器上的文件夾,如 tools

3、修改文件名

需要注意的是,下載的最新的Linux版的CUDNN的文件名是 cudnn-10.0-linux-x64-v7.4.1.5.solitairetheme8,我們需要將其修改爲 .tgz 結尾的文件:

cp cudnn-10.0-linux-x64-v7.4.1.5.solitairetheme8 cudnn-10.0-linux-x64-v7.4.1.5.tgz

4、打開終端,按順序安裝,注意,一定要按順序進行安裝。

tar -xzvf cudnn-10.0-linux-x64-v7.4.1.5.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*

然後安裝其他三個包:

sudo dpkg -i libcudnn7_7.4.1.5-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.4.1.5-1+cuda10.0_amd64.deb 
sudo dpkg -i libcudnn7-doc_7.4.1.5-1+cuda10.0_amd64.deb

5、驗證 CUDNN 是否安裝成功

cp -r /usr/src/cudnn_samples_v7/ .
cd cudnn_samples_v7/mnistCUDNN/
 make clean && make
./mnistCUDNN

如輸出下圖,則說明安裝成功。

如果編譯的時候出現報錯:

./mnistCUDNN: error while loading shared libraries: libcudart.so.10.0: cannot open shared object file: No such file or directory

解決辦法:首先確認/etc/profile中的路徑包含了cuda10.0的安裝路徑及相應的庫文件

export PATH=$PATH:/usr/local/cuda-10.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-10.0/lib64

若無,將上面三行加入profile文件末尾:

sudo vi /etc/profile

使配置文件生效,再次執行。 

source /etc/profile

至此,done.

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章