安裝CUDA® Toolkit 9.0

下載

安裝

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt update
sudo apt install cuda

修改環境變數

編輯/etc/profile檔案，在最後加上下面這一行：

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

驗證安裝成功

重啟之後輸入如下命令：

cat
 
 /proc/driver/nvidia/version

出現如下內容：

輸入如下命令：

nvcc -V

出現如下內容：

安裝cuDNN v7

下載

安裝runtime library

sudo dpkg -i libcudnn7_7.0.4.31-1+cuda9.0_amd64.deb

安裝developer library

sudo dpkg -i libcudnn7-dev_7.0.4.31-1+cuda9.0_amd64.deb 

安裝code samples和cuDNN Library User Guide

sudo dpkg -i libcudnn7-doc_7
 
.0.4.31-1+cuda9.0_amd64.deb

驗證

cp -r /usr/src/cudnn_samples_v7/ $HOME
cd $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
 ./mnistCUDNN

出現test passed資訊。

安裝libcupti-dev library

sudo apt install libcupti-dev

安裝gpu版本的tensorflow

sudo pip install tensorflow-gpu

結束。