Ubuntu16.04.1如何安装TensorFlow1.1.0（GPU版）

• 更新最新的软件包信息

执行命令

sudo apt-get update

• 安装Anaconda（自带Python3.6）

下载 Anaconda，本人下载的是Anaconda3-4.3.1-Linux-x86_64.sh

执行命令

bash Anaconda3-4.3.1-Linux-x86_64.sh

安装过程中，基本都是选择yes，安装成功后，会有当前用户根目录下生成一个anaconda3的文件夹，里面就是安装好的内容

• 安装GPU驱动

首先查看自己的GPU显卡版本

执行命令

lspci

本人的显卡信息如下图：

 

 

在 官网下载相对应版本的显卡驱动

 

 

禁止默认显卡驱动

打开blacklist.conf文件

执行命令

sudo gedit /etc/modprobe.d/blacklist.conf

在里面添加以下内容

blacklist nouveau

安装 kernel-devel（这一步可以跳过！）

在 官网下载三个文件（注意：根据自己系统的版本），如下图：

 

 

然后安装

执行命令

sudo dpkg -i linux-*.deb

安装显卡驱动

首先关闭X-Window

执行命令

sudo service lightdm stop

然后重启系统，通过Ctrl+Alt+F1进入init3（文本模式），登路账户和密码进行显卡驱动安装

执行命令

chmod +x NVIDIA-Linux-x86_64-375.51.run
sudo ./NVIDIA-Linux-x86_64-375.51.run

经过一系列的接受和同意，就可以啦

 

接着开启 X-Window

执行命令

sudo service lightdm start

另外，可通过下面命令查看GPU信息

执行命令

sudo nvidia-xconfig -query-gpu-info

• 安装CUDA

首先在 官网下载

本人下载的是CUDA 8.0

 

然后安装

执行命令

chmod +x cuda_8.0.61_375.26_linux.run
sudo ./cuda_8.0.61_375.26_linux.run

过程一系列的接受和同意就可以。（注意：若缺少什么工具，可根据对应的提示进行安装）

添加环境变量

执行命令

sudo vi /etc/profile

添加内容如下

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

保存并退出

然后使之立刻生效

执行命令

source /etc/profile

添加lib库路径（这一步可以跳过！）

执行命令

sudo vi /etc/ld.so.conf.d/cuda.conf

添加如下内容

/usr/local/cuda-8.0/lib64

保存并退出

然后使之立刻生效

执行命令

sudo ldconfig

验证安装

执行命令

nvcc -V

本人服务器信息如下图

 

 

• 安装cuDNN

首先在 官网下载（注意版本，必须是5.1）

解压

然后依次执行以下命令

 

cd cuda
sudo cp lib64/lib* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/

更新链接

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5
sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5
sudo ln -s libcudnn.so.5 libcudnn.so

• 安装TensorFlow

终于快结束了。。。。。。

首先在官网上下载GPU版TensorFlow（注意：要与 Python版本相对应）

然后执行如下命令

 

sudo pip install tensorflow_gpu-1.1.0rc2-cp36-cp36m-manylinux1_x86_64.whl#md5\=8
37b77c371fc7c04a27923e8f3eda2a6.whl

过程如下图

 

测试TensorFlow

在终端输入python，进入python编译环境

执行命令

import tensorflow as tf

引包tensorflow包，如果没有报错，则安装成功，否则就有问题。

然后输入
 

tf.__version__
tf.__path__

可查看tensorflow的安装版本和安装路径（左右各两根下横线）

最后总结：心累，还是心累。有些坑，一个人摸索了很久才走出来。特此记录，仅供参考，若有问题，敬请指正。之前一直用的是Centos系统上的TensorFlow，和在Ubuntu系统上相比，安装过程还是有些不同。Good luck！