ubuntu16.04安装gpu版本的tensorflow

原来的cpu的tensorflow有点慢，故想安装gpu的tensorflow。安装步骤如下：

1、卸载cpu版本的tensorflow ，sudo pip unistall tensorflow

2、安装cuda，由于本机之前安装过。有点机器安装cuda及nvidia驱动较简单，有点需要拉黑系统自带的显示驱动。

3. 下载cudnn的三个库文件，并安装。

  官网：https://developer.nvidia.com/rdp/cudnn-download，下载runtime library\developer library\

  code samples和cuDNN Library User Guide

  运行：sudo dpkg -i libcudnn6_6.0.21-1+cuda8.0_amd64.deb（对应的包，版本可能不一样）

            sudo dpkg -i libcudnn6-dev_6.0.21-1+cuda8.0_amd64.deb

            sudo dpkg -i libcudnn6-doc_6.0.21-1+cuda8.0_amd64.deb

  安装中途中发现，cudnn 不支持GCC 6之后的版本，由于降低GCC的版本，命令如下：

  运行：sudo apt-get install gcc-6

            sudo apt-get install g++-6

            更改GCC的优先级

          sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-6 60 --slave /usr/bin/g++ g++ /usr/bin/g++-6

4.   安装1.4 版本的gpu tensorflow，安装 pip3 install tensorflow-gpu--1.4 。因为本机原来安装的cuda 是8.0，如果直接pip install tensorflow-gpu，在导入import tensorflow 会报错ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory,关于

这个版本问题可以参考：https://www.jianshu.com/p/d1bb4c694dc0