首頁 > 軟體

Ubuntu 16.04 下 CUDA 8 + cuDNN 5.1安裝

2020-06-16 17:24:40

環境:Ubuntu 16.04 64bit

Nvidia GeForce GTX 1070 (驅動Nvidia-Linux-x86_64-367.57)

安裝:CUDA 8.0.44

cuDNN 5.1

顯示卡驅動安裝:Ubuntu 16.04 Nvidia 顯示卡驅動安裝 見 http://www.linuxidc.com/Linux/2016-12/138907.htm

CUDA是NVIDIA的程式語言平台,想使用GPU就必須要使用cuda。

cuDNN是GPU加速計算深層神經網路的庫。

下載CUDA

去官網https://developer.nvidia.com/cuda-downloads下載CUDA。

我選擇的是:

Operating System : Linux

Architecture : x86_64

Distribution : Ubuntu

Version : 16.04

Installer Type : runfile(local)

deb裡似乎帶了舊顯示卡驅動,安裝完後還要重新裝顯示卡驅動,有些麻煩。就用runfile。

安裝cuda

我安裝的時候執行完,顯示安裝cuda_toolkit的時候庫libGLU.so和libXmu.so缺失。或許可以嘗試先安裝:

apt-get install libglu1-mesa libxi-dev libxmu-dev libglu1-mesa-dev

(見Missing recommended library: libGLU.so)

我是先裝Nvidia驅動的,第一次裝cuda的時候總是報Warning : Incomplete installation ...然後提示沒有驅動。我感覺可能是顯示卡驅動那邊有問題,於是

sudo update-initramfs -u

重新啟動。

sudo modprobe nvidia

然後安裝Cuda。

具體參考:this installation did not install the CUDA driver

安裝:

sudo sh ./cuda_8.0.44_linux.run

(可通過sudo sh ./cuda_8.0.44_linux.run --help檢視一些引數命令,安裝某一部分或是解除安裝的引數都可以看到,可能有些情況安裝要帶--override引數)

一大堆條款什麼的,願意看就看吧,不想看直接回車按到底。

輸入accept接受條款。

接下來只有nvidia驅動那裡不要裝,因為我之前已經裝過了顯示卡驅動。其他都選是或者預設就可以。

流程:

輸入n。不裝nvidia驅動。

輸入y。安裝cuda toolkit

回車。 使用預設安裝路徑/usr/local/cuda-8.0

輸入y。安裝指向/usr/local/cuda的連結

輸入y。安裝Samples。

回車。 使用Samples預設安裝路徑/home/username

環境變數設定

sudo vim ~/.bashrc

在末尾新增

export CUDA_HOME=/usr/local/cuda-8.0

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

export PATH=/usr/local/cuda-8.0/bin:$PATH

儲存退出。

然後重新整理。

source ~/.bashrc

動態連結庫設定

建立檔案:

sudo vim /etc/ld.so.conf.d/cuda.conf

寫入:/usr/local/cuda/lib64

儲存之後使其立即生效:

sudo ldconfig -v

檢查

(1)Cuda版本

nvcc --version

我這裡提示nvcc沒安裝。之前說裝toolkit時缺兩個庫,我裝上後重灌了cuda還是這樣,執行cuda.run --silent --toolkit還是沒用。最後只能sudo apt-get install  nvidia-cuda-toolkit下一個G了,不知道為什麼。

sudo apt-get install  nvidia-cuda-toolkit

nvcc --version

(2)Sample測試

cd /usr/local/cuda/samples

sudo chown -R <username>:<usergroup> .

cd 1_Utilities/deviceQuery

make .

./deviceQuery

輸出一些關於cuda及顯示卡的資訊則正常。make時可能會因gcc、g++版本過高引發問題,但我沒遇到,解決方法可見底部我的參考。

獲取cuDNN

去官網https://developer.nvidia.com/rdp/cudnn-download下載cuDNN,要註冊NVIDIA賬號。

我選擇的是cuDNN v5.1 -> cuDNN v5.1 Library for Linux

安裝cuDNN

下載cuDNN後解壓

cd cuda/include

sudo cp cudnn.h /usr/local/cuda/include #複製標頭檔案

cd ../lib64

sudo rm -rf libcudnn.so libcudnn.so.5    #刪除原動態檔案

sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5  #生成軟連結

sudo ln -s libcudnn.so.5 libcudnn.so      #生成軟連結

若需要更換cudnn版本,則替換原來的libcudnn*,並重新軟連結。

更新連結庫:

sudo ldconfig

Ubuntu 14.04 安裝設定CUDA  http://www.linuxidc.com/Linux/2014-10/107501.htm

Ubuntu 12.04設定NVIDIA CUDA5.5實錄  http://www.linuxidc.com/Linux/2014-10/107502.htm

Ubuntu安裝Theano+CUDA  http://www.linuxidc.com/Linux/2014-10/107503.htm

關於Ubuntu 12.04 下 CUDA5.5 的安裝請參看如下連結 Ubuntu 12.04 安裝 CUDA-5.5

Caffe設定簡明教學 ( Ubuntu 14.04 / CUDA 7.5 / cuDNN 5.1 / OpenCV 3.1 ) http://www.linuxidc.com/Linux/2016-09/135016.htm

在Ubuntu 14.04上設定CUDA+Caffe+cuDNN+Anaconda+DIGITS  http://www.linuxidc.com/Linux/2016-11/136775.htm


IT145.com E-mail:sddin#qq.com