记录在鲲鹏服务器上使用bazel 变异tensorflow1.13cpu版

记录在鲲鹏服务器上使用bazel 变异tensorflow1.13cpu版

一、准备安装包
tensorflow1.13
bazel-0.19.2-dist.zip(bazel对tensorflow版本有要求,所以版本必须是19.2)
protobuf-3.6.1
nccl
(需要百度链接可以联系我)
预先准备的安装包
二.安装bazel
安装需要的环境centos系统(不过这里鲲鹏服务器是redhat)。

yum install -y java-1.8.0-openjdk java-1.8.0-openjdk-devel
yum groupinstall "Development Tools"
yum install -y gcc g++ kernel-devel

解压bazel-0.19.2-dist.zip,并进入bazel文件夹。

unzip bazel-0.19.2-disk.zip
cd bazel

没有unzip可以装一下。

yum install  unzip

执行安装脚本

env EXTRA_BAZEL_ARGS="--host_javabase=@local_jdk//:jdk" bash ./compile.sh

添加环境变量可查看,linux添加环境变量和依赖库
查看是否安装成功执行

bazel version

如下图所示证明安装成功。
在这里插入图片描述

三、安装protocbuf

解压并进入文件夹

tar -xvzf protocbuf-3.6.1.tar.gz
cd protocbuf-3.6.1

执行编译脚本

./autogen.sh
./configure

编译并安装

make -j8
make check
make install
ldconfig

检查是否安装成功

protoc --version

出现如下信息即为安装成功。
在这里插入图片描述

四、编译tensorflow-cpu版

解压并进入tensorflow

unzip tensorflow-1.13rc.zip
cd tensorflow-1.13rc

执行configure.py文件

python configure.py

CPU版选项
在这里插入图片描述
若编译GPU版在
在这里插入图片描述
选择Yes,并填写cuda路径和cudnn路径即可,前提为以安装显卡驱动和CUDA以及cudnn
在这里工作打断了一下暂停

猜你喜欢

转载自blog.csdn.net/qq_39327033/article/details/110422381