【TensorFLow】Ubuntu16.04安裝CUDA8.0+cuDNN6.0+TensorFlow

NVIDIA驅動

---

Ubuntu在安裝NVIDIA驅動前要先將BIOS中的的Secure Boot設定為Disabled

附加驅動安裝

最簡單的安裝方式是直接在“系統設定->軟體和更新->附加驅動”中安裝

終端安裝

如果需要在終端中安裝，可以按照以下步驟進行

1. 刪除舊NVIDIA驅動

sudo apt-get purge nvidia*

2. 禁用nouveau驅動

開啟黑名單檔案blacklist.conf

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

在末尾新增

blacklist vga16fb 
blacklist nouveau 
blacklist rivafb 
blacklist rivatv 
blacklist nvidiafb

然後更新核心

sudo update-initramfs -u

重啟系統確認nouveau是否已經被遮蔽掉

lsmod | grep nouveau

3. 安裝NVIDIA驅動

新增Graphic Drivers PPA

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

尋找合適的驅動版本

ubuntu-drivers devices

按ctrl+alt+F1進入tty1模式，關閉(圖形)桌面顯示管理器LightDM

sudo service lightdm stop

安裝nvidia driver，安裝完成後重啟

sudo apt-get install nvidia-384
sudo reboot

重啟系統後，執行下面的命令檢視驅動的安裝狀態顯示安裝成功

sudo nvidia-smi

CUDA

---

首先在官網下載CUDA8.0

我下載的是deb包，所以直接安裝就行了

$ sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
$ sudo apt-get install cuda

如果是run檔案，則執行即可，中間過程中注意不要選擇安裝NVIDIA驅動，其它都可以選擇yes

sudo sh cuda_8.0.44_linux.run

然後開啟配置檔案profile

$ sudo vim /etc/profile

在末尾新增以下環境變數

# setting the environment variables so CUDA will be found
# After open profile, Add follow code at the end of file
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

然後更新環境變數

$ . /etc/profile

cuDNN

---

首先在官網下載cuDNN6.0，需要登入填寫一個調查問卷

然後進入下載目錄進行解壓

tar zxvf  cudnn-8.0-linux-x64-v6.0.tgz

最後複製檔案到CUDA的安裝目錄即可

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

TensorFLow

---

安裝方法可以參考極客學院-下載與安裝

我用Python3安裝的TensorFlow

sudo pip3 install -U --pre setuptools wheel
sudo pip3 install -U --pre numpy scipy matplotlib scikit-learn scikit-image
sudo pip3 install -U --pre tensorflow-gpu

想要加快速度的話，請參照這篇部落格這裡使用國內映象進行安裝。

安裝完畢後，在終端中輸入python（或python3），然後輸入：

import tensorflow

沒有報錯說明你已經可以正常使用了