ubuntu安装TensorFlow（支持不同版本）

CUDA+CUDNN准备：

1.安装NVIDIA驱动

（1）查询NVIDIA驱动

首先去官网(http://www.nvidia.com/Download/index.aspx?lang=en-us)查看适合自己显卡的驱动（下载runfile文件）： 

2）安装驱动

先按Ctrl + Alt + F1到控制台，关闭当前图形环境 

　　$sudo service lightdm stop

卸载可能存在的旧版本 nvidia 驱动

sudo apt-get remove --purge nvidia*

安装驱动可能需要的依赖 
$sudo apt-get update
$sudo apt-get install dkms build-essential linux-headers-generic
 sudo nano /etc/modprobe.d/blacklist-nouveau.conf

 在文件 blacklist-nouveau.conf 中加入如下内容：
blacklist nouveau
options nouveau modeset=0
保存退出，执行
 sudo update-initramfs -u

 

ctrl+O保存。然后回车 ctrl+X 退出
reboot重启

运行驱动安装文件

重启后再次进入字符终端界面（Ctrl + Alt + F1），并关闭图形界面

  $sudo service lightdm stop

安装驱动

  $sudo chmod u+x NVIDIA-Linux-x86_64-361.45.11.run

  $sudo ./NVIDIA-Linux-x86_64-361.45.11.run（sudo ./NVIDIA-Linux-x86_64-375.20.run –no-opengl-files）
或 sudo ./NVIDIA-Linux-x86_64-375.20.run –no-x-check –no-nouveau-check –no-opengl-files

kenel 不装选Yes； X-config Yes

最后重新启动图形环境 

$sudo service lightdm start

可以通过以下命令确认驱动是否正确安装

$cat /proc/driver/nvidia/version

2.安装cuda8.0

9.0的时候用runfile没有安装成功，用安装包装的，ok了。安装包本地：

`sudo dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb`
`sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub`
`sudo apt-get update`
`sudo apt-get install cuda`

可能出现 Driver/library version mismatch 的问题，重启，或者按照此方法。

需要进入安全模式安装（lightdm），在安装cuda时是可以自动安装NVIDIA驱动的，但是这样安装的驱动版本较低，且可能有问题，所以需要先安装NVIDIA驱动，在安装CUDA时选择不安装驱动。

切换到cuda_8.0.44_linux.run的目录，执行

$ sudo sh cuda_8.0.44_linux.run

安装CudNN

下在对应的版本（cudnn-9.2-linux-x64-v7.1）按照官网教程，解压复制：

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*

 

正式安装：

一、查看系统所安装的python版本

打开终端输入指令：python，如图1所示，我的系统是ubuntu16.04.03，默认安装的python版本为2.7.12。

二、安装python对应版本的pip和依赖包

若python版本为2.7，则输入如下命令：
sudo apt-get install python-pip python-dev

若python版本为3.x，则输入如下命令：
sudo apt-get install python3-pip python3-dev

三、升级pip版本

在装tensorflow之前，不管是不是最新的pip版本，都要更新一下，具体命令如下：
python 2.7版本：sudo pip install --upgrade pip
python 3.x版本：sudo pip3 install --upgrade pip

四、更改pip源地址（提高下载速度）

修改 ~/.pip/pip.conf (没有就创建一个文件夹及文件，文件夹要加"."，表示是隐藏文件夹)，内容如下：
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host=mirrors.aliyun.com

五、安装TensorFlow

TensorFlow可以安装CPU和GPU两种版本，CPU版本安装命令如下：
python 2.7版本：sudo pip install tensorflow
python 3.x版本：sudo pip3 install tensorflow

GPU版本安装命令如下：
python 2.7版本：sudo pip install tensorflow-gpu
python 3.x版本：sudo pip3 install tensorflow-gpu

若上述命令执行过程没有报错，则安装成功。
六、测试安装结果

进入python编译环境，导入TensorFlow，做一个简单的加法运算，如图2所示。