Ubuntu16.04安裝Nvidia顯卡驅動+Cuda8.0+Cudnn6.0

一、安裝Nvidia顯卡驅動（gtx1050ti）

參考鏈接：Ubuntu16.04.2 LTS 64bit系統裝機記錄中的顯卡驅動安裝部分。

 

二、安裝Cuda8.0

 

1、確定自己的系統信息，以Ubuntu16.04.2 LTS 64bit為例，到官網https://developer.nvidia.com/cuda-downloads下載相應的安裝文件

2、下載后在當前目錄運行命令安裝

sudo ./cuda_8.0.61_375.26_linux.run.26_linux-run
安裝過程中基本上選擇默認路徑以及選Yes，但是提示是否安裝Nvidia驅動時一定選擇No，因為此前已經安裝了相應的顯卡驅動

 

3、安裝成功后，聲明一下環境變量，並將其寫入到 ~/.bashrc 的尾部:

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

 

4、設置環境變量

終端輸入：sudo gedit /etc/profile
文件末尾加入：export PATH=“/usr/local/cuda/bin:$PATH”
運行：source /etc/profile，查看是否有誤

 

5、設置動態鏈接庫

終端輸入：sudo gedit /etc/ld.so.conf.d/cuda.conf
添加：/usr/local/cuda/lib64
執行生效：sudo ldconfig

 

6、測試cuda的samples

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
make
sudo ./deviceQuery

 

7、完畢

 

三、安裝Cudnn6.0

 

1、確定已經安裝的Cuda版本，以Cuda8.0為例，到官網https://developer.nvidia.com/rdp/cudnn-download下載相應的庫文件（下載前需要登錄網站，因此需要進行注冊）

2、下載后進行解壓

3、進入include文件夾，執行

sudo cp cudnn.h /usr/local/cuda/include/  #復制頭文件

 

4、進入lib64文件夾，執行

sudo cp lib* /usr/local/cuda/lib64/  #復制動態鏈接庫

 

5、執行

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.6           #刪除原有動態文件
sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6     #生成軟鏈接
sudo ln -s libcudnn.so.6 libcudnn.so            #生成軟鏈接

 

注意！！軟鏈接中的數字6.0.21等參照文件夾lib64中原有的軟鏈接名

6、完畢