深度學習GPU環境搭建

• ubuntu16.04+nvidia驅動+cuda9.0+cudnn7.0.5
• 安裝nvidia驅動
• 安裝cuda
• 安裝 cudnn：

ubuntu16.04+nvidia驅動+cuda9.0+cudnn7.0.5

安裝nvidia驅動

sudo vim /etc/modprobe.d/blacklist.conf
blacklist nouveau
options nouveau modeset=0

sudo update-initramfs -u

reboot

lsmod | grep nouveau
進入命令列介面 ctrl+alt+f1，開始安裝

sudo service lightdm stop
cd install_package
sudo chmod 777 NVIDIA-Linux-x86_64-390.67.run
sudo ./NVIDIA-Linux-x86_64-390.67.run

reboot

檢查gpu安裝情況
nvidia-smi

安裝cuda

sudo chmod 777 cuda_9.0.176_384.81_linux.run

sudo ./cuda_9.0.176_384.81_linux.run

sudo gedit /etc/profile

開啟檔案後在檔案末尾新增路徑，也就是安裝目錄，命令如下：

export PATH=/usr/local/cuda-9.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH　
source /etc/profile

sudo gedit /etc/ld.so.conf.d/cuda.conf
在開啟的檔案中新增如下語句：
/usr/local/cuda/lib64
執行sudo ldconfig

安裝 cudnn：

tar -zxvf cudnn-9.0-linux-x64-v3.0.29.tgz
cd cuda
sudo cp lib64/lib* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/

cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.7.3.0 # 自己檢視.so的版本
sudo ln -sf libcudnn.so.7.3.0. libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig