這次介紹的是使用 tensorflow1.8, cuda9.0, cudnn7.0的版本

• https://developer.nvidia.com/cuda-90-download-archive 下載相應的cuda，推薦使用deb進行安裝

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

• https://developer.nvidia.com/cudnn下載cudnn

tar -xzvf cudnn-9.0-linux-x64-v7.tgz
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*

• 配置環境變量，在～/.bashrc當中加入下面兩句

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64$LD_LIBRARY_PATH
 

• 安裝tensorfloww-gpu，還有nvidia庫完事

sudo apt-get install nvidia-cuda-toolkit
pip install tensorflow-gpu

ubuntu16.04 CUDA, CUDNN 安裝