Centos 7.4 源码安装GPU版Tensorflow

 GPU版Tensorflow需要nvidia driver以及cuda和cuDNN的支持，请先安装以上支持。相关安装过程可参考本人博客：https://blog.csdn.net/u013431916/article/details/79656958

 

 

• 安装jdk：

1 在oracle官网下载安装包：

http://www.oracle.com/technetwork/java/javase/downloads/jdk8-downloads-2133151.html

2 用rpm安装：

$ sudo rpm -i jdk-8u161-linux-x64.rpm

3 配置环境变量（具体jdk版本号需要进入/usr/java文件夹查看）：

$ export JAVA_HOME=/usr/java/jdk1.8.0_161

4 验证安装：

$ java -version

 

 

 

 

• 安装bazel：

1 下载指定版本的安装脚本：

https://github.com/bazelbuild/bazel/releases

2 修改文件属性：

$ chmod +x bazel--installer-linux-x86_64.sh

3 安装：

$./bazel--installer-linux-x86_64.sh --user

4 配置PATH变量：

$ vim ~/.bashrc

添加：

export PATH="$PATH:$HOME/bin"

然后：

$ source ~/.bashrc

 

 

• 安装tensorflow：

安装NCCL 2.0教程见：Ubuntu 安装 NCCL 2.0

1 安装依赖：

$ sudo yum -y install numpy swig python-devel python-wheel python-pip zlib zlib-devel

2 在github上下载安装文件.zip，并解压：

https://github.com/tensorflow/tensorflow

$ unzip tensorflow*.zip

3 进入解压后的文件夹进行配置：

$ cd tensorflow
$ ./configure

参考配置如下：

4 编译（gpu版），这个过程有点长：

$ bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

5 生成.whl包：

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

6 安装生成的包：

$ pip install /tmp/tensorflow_pkg/tensorflow-*.whl

 

8 命令行输入“python“，在python环境下执行“import tensorflow”，如无报错，则安装成功。

参考：

1. https://community.mellanox.com/docs/DOC-2911#jive_content_id_Required_Software

2. https://docs.bazel.build/versions/master/install.html