bazel源码编译Tensorflow

本文档详细记录了如何使用bazel从源码编译TensorFlow的libtensorflow_cc.so和libtensorflow_framework.so库。首先,确保硬件满足GCC4.8以上和bazel0.10.0以上的要求。接着,安装jdk1.8,然后安装bazel,通过./compile.sh编译。之后,安装tensorflow r1.8,编译过程中可能遇到错误,如“clock_gettime”,需要修改tensorflow.bzl文件并调整编译命令。经过长时间编译,最终会在bazel-bin目录下得到所需的.so库,整个过程充满挑战。
摘要由CSDN通过智能技术生成

因为研究需求,要从Tensorflow源码编译libtensorflow_cc.so和libtensorflow_framwork.so两个库,工具是bazel。

编译硬件需求:GCC4.8以上,bazel0.10.0以上,tensorflow r1.8。

我的硬件需求:GCC7.3.0,bazel0.15.0,tensorflow r1.8。

GCC安装就不用说了,一般Linux自带,gcc --version如果版本低要编译一份新的。

1.安装jdk

安装bazel需要jdk1.8版本的。

sudo apt-get install openjdk-8-jdk

一定是1.8版本的!!!安装完要设置环境变量:

通过java -version查看版本。

2.安装bazel

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

好的,以下是在Linux系统中从源码编译安装TensorFlow 1.13.1的步骤: 1. 安装必要的依赖项: ``` sudo apt-get update && sudo apt-get install python3-dev python3-pip python3-wheel python3-venv sudo apt-get install libhdf5-dev libc-ares-dev libeigen3-dev sudo apt-get install libatlas-base-dev libopenblas-dev libblas-dev sudo apt-get install liblapack-dev cython3 ``` 2. 创建Python虚拟环境: ``` python3 -m venv tensorflow_env source tensorflow_env/bin/activate ``` 3. 下载TensorFlow 1.13.1源代码: ``` wget https://github.com/tensorflow/tensorflow/archive/v1.13.1.tar.gz tar -xzvf v1.13.1.tar.gz ``` 4. 进入TensorFlow源代码目录: ``` cd tensorflow-1.13.1 ``` 5. 配置编译选项: ``` ./configure ``` 按照提示进行配置,其中需要注意的是: - 选择Python解释器的路径为虚拟环境中的Python路径,例如:/home/user/tensorflow_env/bin/python3.6; - 如果您的计算机支持AVX指令集,请在编译选项中启用该选项,以获得更好的性能。 6. 开始编译: ``` bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package ``` 编译过程可能需要一些时间,请耐心等待。 7. 生成安装包: ``` ./bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg ``` 此步骤将会在/tmp/tensorflow_pkg目录下生成TensorFlow 1.13.1的安装包。 8. 安装TensorFlow: ``` pip install /tmp/tensorflow_pkg/tensorflow-1.13.1-*.whl ``` 9. 验证TensorFlow是否安装成功: ``` python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))" ``` 如果输出了一个随机数的和,则说明TensorFlow安装成功。 希望以上步骤能够帮助您成功安装TensorFlow 1.13.1。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Code Wang

打赏后可联系博主分享本博客内容

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值