记录在鲲鹏服务器上使用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
在这里工作打断了一下暂停