ubuntu16.04在GTX1070環境下安裝 cuda9.1

設備要求

• 系統：Ubuntu16.04

• 顯卡：GTX 1070

• 驅動：nvidia系列，顯卡驅動的版本必須大於等於cuda的sh文件名裏面的版本號

  驅動可從 此處 下載，我已經整理好了

檢查安裝驅動

nvidia-smi

顯示以下信息

若是徹底無反應或者驅動型號小於cuda的sh文件名裏面的版本號則須要從新安裝驅動。
首先去官網下載符合GPU的驅動。

一、刪除舊的驅動

sudo apt-get purge nvidia*

二、禁用自帶的nouveau nvidia驅動

看下nouveau 是否已經被禁用

lsmod | grep nouveau

若是沒有任何顯示說明表示已經被禁用，不然繼續下面操做
修改/etc/modprobe.d/blacklist.conf，添加如下內容

blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb

新建blacklist-nouveau.conf文件，

sudo vim /etc/modprobe.d/blacklist-nouveau.conf #建立一個文件（注：按一下i鍵，表示如今進行內容插入）

並添加以下內容：

blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off

再更新一下：

sudo update-initramfs –u

確認下Nouveau是已經被禁用:

lsmod | grep nouveau 

沒有輸出什麼東西，說明已經成功關閉了。

三、關閉X-window服務

Ctrl+Alt+F1切換到無桌面命令終端：這裏lightdm是你本身的顯示管理器，也多是gdm，kdm，究竟是哪個可使用 cat /etc/X11/default-display-manager進行查看，而後修改，並關閉顯示管理器。

sudo service lightdm stop

此時正式進入終端界面：

Login:用戶帳號
Password:用戶密碼

安裝，注意：這裏的參數（–no-opengl-files）很關鍵:

cd  /home/xxx/tmp # 導到你的下載的驅動放在哪裏的地址
sudo sh NVIDIA-Linux-x86_64-418.56.run --no-opengl-files

按照以下步驟安裝:

(1)accept

(2)contiuned install

後面默認yes安裝就行了

啓動顯示器：（lightdm只是個人顯示管理器，你多是前面的說的gdm）

sudo service lightdm start

而後按Ctrl+Alt+F7 進入到桌面進行操做

檢查是否成功

nvidia-smi

這裏會顯示顯卡信息

驅動已經正常安裝了，能夠進行下面操做了。

安裝cuda9.1

首先去官網下載cuda9.1， 下載那個1.6G的.run文件，下載完畢就能夠正式安裝了。

進入下載目錄，給文件添加運行權限：

chmod +x ./cuda_9.1.85_387.26_linux.run

運行安裝

sudo ./cuda_9.1.85_387.26_linux.run

啓動安裝程序，一直按空格到最後（能夠選擇Ctrl+c跳過），不用擔憂，到99%的時候，輸入accept接受條款

注意：第一個提醒你是否安裝驅動時，選「n」，其他都「y」

安裝完畢後就須要添加環境了，這步很重要！！！

gedit ~/.bashrc

把下面的內容添加到最後：

export CUDA_HOME=/usr/local/cuda

export PATH=$PATH:$CUDA_HOME/bin

export LD_LIBRARY_PATH=/usr/local/cuda-9.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

測試是否安裝成功

nvcc -V

若是你看到如下顯示，cuda9.1 已經裝好了！

安裝cudnn7.x

註冊一個帳號，去官網上下載cudnn7.x。

選擇適配Linux的，相似 cuDNN v7.0.5 Library for Linux

下載完直接解壓，解壓會出現一個cuda文件夾，裏面有兩個文件include 和 lib64，把裏面的文件copy到/usr/local/cuda/裏面相應的目錄裏。 若是你就在local下解壓的就不要移動了。只須要給文件加讀權限便可！

sudo chmod a+x /usr/local/cuda/include/cudnn.h
sudo chmod a+x /usr/local/cuda/lib64/libcudnn*

而後更新網絡鏈接：

cd /usr/local/cuda/lib64/  
sudo chmod +r libcudnn.so.7.0.5  # 本身查看.so的版本  
sudo ln -sf libcudnn.so.7.0.5. libcudnn.so.7  
sudo ln -sf libcudnn.so.7 libcudnn.so  
sudo ldconfig 

查看cudnn版本，檢查是否安裝好：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

若是顯示相似以下信息，表示安裝成功

卸載CUDA以及CUDNN

（1）卸載CUDA

sudo /usr/local/cuda-9.0/bin/uninstall_cuda_9.0.pl 

（2）卸載CUDNN，刪除原來的cudnn文件：

sudo rm -rf /usr/local/cuda-9.0/lib64/libcudnn*
sudo rm -rf /usr/local/cuda-9.0/include/cudnn.h

參考：

[1] Ubuntu16.04下cuda9.0+cudnn7.0安裝指南
[2] Ubuntu16.04 + GTX1080 解決桌面重複登陸問題