如何在ubuntu16.04安装TensorflowGPU

我这里配置的是tensorflow1.6.0 + cuda9.0 + cudann 7.1.1

1.更改驱动
在“软件和更新”里找到 附加驱动把驱动换成NIVIDA BINARY 我的是384.111

2.安装cuda https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal 这里是cuda9.0的下载地址，英伟达有对应的安装指令 输入即可
3.安装cudann
登录后在这里下载 对应打这个

cuDNN v7.1.1 Library for Linux
用tar指令解压

然后
sudo cp cuda/ include/cudnn.h /usr/local/cuda/include
sudo cp cuda/ lib64/libcudnn*  /usr/local/cuda/lib64
最后sudo gedit  ~/.bashrc
加上
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"
export CUDA_HOME=/usr/local/cuda

export PATH=/usr/local/cuda/bin:$PATH
4.装好tensorflowGPU环境 就好了
视频教程： https://www.youtube.com/watch?v=fY0n0WD3UXs