ubuntu18.04 安裝tensorflow-gpu

目錄驅動
CUDA Toolkit 9.0
CUDNN 7.0
libcupti
配置
安裝tensorflow-gpu
測試
主要參考：https://medium.com/@taylordenouden/installing-tensorflow-gpu-on-ubuntu-18-04-89a142325138
驅動
https://blog.csdn.net/tonydz0523/article/details/83109343 

安裝完成後最好重啟下。
檢查驅動是否安裝成功

CUDA Toolkit 9.0
目前tensorflow只支持CUDA Toolkit 9.0。
下載地址：
https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1704&target_type=runfilelocal
目前沒有支持ubuntu18.04的CUDA Toolkit 9.0。選擇17.10的版本，安裝base installer一般夠用了。

sudo chmod +x cuda_9.0.176_384.81_linux.run
./cuda_9.0.176_384.81_linux.run override

installing with an unsupported configuration？時選擇yes
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81？為了穩定最好選擇no
樣例可以不安裝。
CUDNN 7.0
地址：https://developer.nvidia.com/rdp/cudnn-archive

# 解壓
tar -zxvf cudnn-9.0-linux-x64-v7.tgz
# 複制相應文件
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/




sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include/
# 所有用戶可讀
sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

libcupti

sudo apt-get install libcupti-dev

配置
在~/.bashrc中加入

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

安裝tensorflow-gpu
根據需要進行修改

pip install upgrade tensorflow-gpu

測試

成功!