【Tensorflow】Ubuntu 安装 Tensorflow gpu

安装环境：Ubuntu 16.04lts 64位，gcc5.4

1.安装Cuda

1. 下载cuda toolkit。

　　下载cuda8.0 地址：https://developer.nvidia.com/cuda-downloads 版本选择：Operating System：Linux; Architecture : x86_64; Distribution : Ubuntu; Version : 16.04; Installer Type : runfile(local)。只下载Base Installer就可以了。

2. 禁用 nouveau

　　终端运行：

lsmod | grep nouveau

　　若有输出则代表nouveau正在加载。

　　禁用方法：在/etc/modprobe.d中创建文件blacklist-nouveau.conf，在文件中输入以下两行内容

blacklist nouveau
options nouveau modeset=0   

　　终端输入

sudo update-initramfs –u 

　　设置完毕可以再次运行 lsmod | grep nouveau查看nouveau是否正在加载。如果仍输出信息，则重启电脑。没有输出信息则代表禁用成功。

3. 安装cuda toolkit

7. Configure the installation

　　切换目录到tensorflow安装目录下，会有一个configure文件，输入命令

./configure

　　

8. Build the pip package

输入命令

bazel build --config=opt --config=cuda --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" //tensorflow/tools/pip_package:build_pip_package

   命令 --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" 是为了gcc兼容性。

 

bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

　　上述命令执行完毕后，在/tmp/tensorflow_pkg directory目录下会有.whl 类型的文件，执行命令：

9.Install the pip package

　　执行命令，注意更换文件名：

sudo pip install /tmp/tensorflow_pkg/tensorflow-1.3.0-py2-none-any.whl