1. 程式人生 > >ubuntu16.04+1080顯示卡+cuda8.0+cudnn5.1安裝記錄

ubuntu16.04+1080顯示卡+cuda8.0+cudnn5.1安裝記錄

1. 顯示卡驅動安裝

(1)在nvidia官網上下載相應的驅動:NVIDIA-Linux-x86_64-384.111.run

(2)解除安裝可能存在的舊版本 nvidia 驅動(對沒有安裝過 nvidia 驅動的主機,可省略)

在終端輸入:

sudo apt-get remove --purge nvidia*

(3)安裝驅動可能需要的依賴

sudo apt-get update

sudo apt-get install dkms build-essential linux-headers-generic

(4)把 nouveau 驅動加入黑名單

sudo nano /etc/modprobe.d/blacklist-nouveau.conf

在檔案 blacklist-nouveau.conf 中加入如下內容:

   blacklist nouveau

   blacklist lbm-nouveau

   options nouveau modeset=0

   alias nouveau off

   alias lbm-nouveau off

(5)禁用 nouveau 核心模組

echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf

sudo update-initramfs -u

(6)重啟
(7)重啟後再次進入字元終端介面,並關閉圖形介面(ALT+CTRL+F1)

sudo service lightdm stop

(8)安裝驅動

sudo chmod u+x NVIDIA-Linux-x86_64-384.111.run

 sudo ./NVIDIA-Linux-x86_64-384.111.run

(9)重啟

(10)重啟後,在終端輸入:

nvidia-smi

顯示以下資訊即正確安裝:

2. cuda8.0安裝

從NVIDIA下載cuda安裝檔案:cuda_8.0.61_375.26.linux.run

(1)

sudo sh cuda_8.0.61_375.26_linux.run

會出現一串須知一樣的資訊,可以直接ctrl+c跳過,也可按空格閱讀

閱讀完資訊後,會有一些選擇,需要注意的是,是否安裝NVIDIA驅動問題時,應選擇否。其他選擇yes或者預設即可。

安裝成功後會出現如下介面:

===========

= Summary =

===========

Driver: Not Selected

Toolkit: Installed in /usr/local/cuda-8.0

........

***WARNING: Incomplete installation! Tis installation did not install the CUDA Driver. A driver ......

.....

Logfile is /opt/temp//cuda_install_6583.log

安裝完畢後,需宣告環境變數,並將其寫入到~/.bashrc的尾部:

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"

export CUDA_HOME=/usr/local/cuda

(2)測試cuda

cd  ~/NVIDIA_cuda-8.0_Samples/1_Utilities/deviceQuery

make 

./deviceQuery

出現以下結果即是安裝成功:

3. 安裝cuDNN

下載cuDNN檔案:cudnn-8.0-linux-v5.1.tgz並解壓

(1)複製標頭檔案

cd ~/cuda/include

sudo cp cudnn.h /usr/local/cuda/include/

(2)lib64目錄下動態檔案進行復制和連線

cd ~/cuda/lib64

sudo cp lib* /usr/local/cuda/lib64/  #複製動態連結庫

cd /usr/local/cuda/lib64/

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

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

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