1. 程式人生 > >Ubuntu 18.04 安裝 Tensorflow 環境之三:安裝 CUDNN 7.4.1

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

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

安裝過程

1、下載CUDNN

NVIDIA官網上下載 NVIDIA CUDNN 相關資源,NVIDIA 官網需要註冊,在 login 進去之後,點選同意“”,相關資源才會顯示。

在官網上下載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.