Ubuntu CUDA+cudnn+驅動安裝

• 安裝Ubuntu
• 安裝CUDA
• 安裝驅動
• 安裝CUDNN

tensorflow的要求版本可能會變化，一定要按照官網上建議的裝GPU support

安裝Ubuntu

最新版本是18.04 但cuda那邊並不支援18 所以我安裝的是16.04版本 安裝可以參考Ubuntu雙系統安裝

安裝CUDA

一定要安裝 9.0 的 下載地址
然後選擇 deb的network版本 下載好之後複製到 home 主目錄下
之後執行下面的命令

sudo apt-get install cuda-9.0

然後中途可能會出現一個對話方塊 我選的是YES 接著隨便輸個密碼就行 其他預設yes就行

然後配置環境變數

gedit ~/.bashrc

加入

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/lib/nvidia-390"
export CUDA_HOME=/usr/local/cuda
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export PATH=/usr/lib/nvidia-390/bin${PATH:+:${PATH}}
 

其中版本號（也就是數字的地方）改一下

重新整理環境變數

source ~/.bashrc

測試一下nvcc

nvcc -V

沒蹦出提示錯誤提示就是成功啦

接著試試跑樣例

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

如果跳出你的顯示卡資訊就是成功啦 ， 如果失敗了就接著看下去

安裝驅動

這一步僅僅針對nvcc-V成功 但 樣例失敗的情況(就是這個浪費了我兩天)
首先 去英偉達官網下載對應的驅動下載地址 tensorflow對驅動有版本要求，所以要先去

禁用secure boot
在BIOS中將secure boot設定disable

禁用nouveau第三方驅動

sudo gedit /etc/modprobe.d/blacklist.conf

在最後新增

blacklist nouveau

更新核心

sudo update-initramfs -u

重啟後 輸入

lsmod | grep nouveau

沒有輸出就大功告成啦
按Ctrl + Alt + F2 重新輸入你的使用者名稱 密碼 (最好先拍張照哦，按了之後就看不見你的瀏覽器了)

接著按執行下面程式碼

service lightdm stop
sudo chmod a+x NVIDIA-Linux-x86_64-xxx.xx.run
sudo ./NVIDIA-Linux-x86_64-xxx.xx.run –no-opengl-files

會有幾執政個選項 直接預設就行了

等他執行完了之後重啟就OK 不會重啟的直接按電源
然後執行上面出錯的地方 就可以啦(反正我是可以的)

安裝CUDNN

下載地址
這裡要下載那個cuda9.0的linux檔案 就是下面這個


下載下來會是一個tgz檔案
解壓他到local資料夾下

tar -zvxf 檔名.tgz -C /usr/local

sudo chmod a+r /usr/local/cuda/include/cudnn.h  

執行完之後

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

有輸出就是成功安裝cudnn了


至此cuda和cudnn就安裝成功了


有不對的地方歡迎大佬指正

參考：
Linux安裝NVIDIA顯示卡驅動的正確姿勢
最正確的姿勢安裝cudnn，網上大多數教程都太坑了