搞機器學習也有一段時間了，每次部署GPU開發環境就是一場戰爭，先記錄一下基本步驟，結合網上資料和個人實踐整理如下：

1、檢查BIOS啟動項，關閉一些選項

在開機啟動項的Security選項中檢查UEFI是否開啟，如果開啟的話請立馬關掉它（重要）
在開機啟動項的Boot選項中檢查Secure Boot是否開啟，如果開啟的話請立馬關掉它（重要）

2、安裝相關依賴

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
 

sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

如果有圖形化介面則需要禁用x-window服務

sudo service lightdm stop
或
sudo /etc/init.d/lightdm stop

3、禁用 nouveau

安裝好依賴包後需要禁用 nouveau，只有在禁用掉 nouveau 後才能順利安裝 NVIDIA 顯示卡驅動，禁用方法就是在 sudo vim /etc/modprobe.d/blacklist.conf檔案中新增一條禁用命令

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

執行如下命令：

echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf build the new kernel by:

最後更新後重啟電腦

sudo update-initramfs -u
reboot

4、查詢NVDIA驅動

首先去官網 http://www.nvidia.com/Download/index.aspx?lang=en-us

5、安裝CUDA庫

進入 https://developer.nvidia.com/cuda-downloads，依次選擇 CUDA 型別然後下載對應的CUDA即可，下面以CUDA8.0為例：

如果有圖形化介面則需要禁用x-window服務
安裝cuda驅動

sh cuda_8.0.61_375.26_linux.run

注：
這個時候會有很多提示需要你確認，由於在第一步已經成功安裝了顯示卡驅動，所以這裡就不需要再次安裝了，因此 ，Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 3XX.XX ?這裡選擇 no，其餘都是yes或者accept

在.bashrc新增環境變數後source .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}}

6、驗證CUDA是否安裝成功

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

若看到類似以下資訊則說明 cuda 已安裝成功：

7、安裝cuDNN

下載後進行解壓：

tar -zxvf cudnn-8.0-linux-x64-v5.0-ga.tgz

進入cuDNN5.0解壓之後的include目錄，在命令列進行如下操作：

cd cuda/include
sudo cp cudnn.h /usr/local/cuda/include  #複製標頭檔案

再將進入lib64目錄下的動態檔案進行復制和連結：

cd ..
cd lib64
sudo cp lib* /usr/local/cuda/lib64/    #複製動態連結庫
cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.5.0.5
sudo ln -sf libcudnn.so.5.0.5 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
sudo ldconfig