Ubuntu安装编译TensorFlow C++

1、下载TensorFlow源码

https://github.com/tensorflow/tensorflow

2、安装编译protobuf

protobuf版本要和TensorFlow版本对应;

protobuf版本查看方法:

打开TensorFlow源码下tensorflow-master/tensorflow/workspace2.bzl文件

搜索可以找到地址"https://github.com/protocolbuffers/protobuf/archive/v3.9.2.zip"

安装编译:

cd protobuf-XXX/
./autogen.sh
./configure
 make
 make check
 sudo make install
 sudo ldconfig # refresh shared library cache.

查看protobuf版本

protoc --version

3、安装编译bazel

bazel版本要和TensorFlow版本对应;

bazel版本查看方法:

打开TensorFlow源码下tensorflow-master/configure.py

搜索可以找到:

_TF_MIN_BAZEL_VERSION = '4.2.2'

_TF_MAX_BAZEL_VERSION = '4.99.0'

这里可以选择安装4.2.2版本

安装编译:
下载:链接 Releases · bazelbuild/bazel · GitHub

安装依赖:

sudo apt-get install pkg-config zip g++ zlib1g-dev unzip python

 编译:

chmod +x bazel-<version>-installer-linux-x86_64.sh
./bazel-<version>-installer-linux-x86_64.sh --user

4.2.2版本对应写法:

chmod +x bazel-4.2.2-installer-linux-x86_64.sh
./bazel-4.2.2-installer-linux-x86_64.sh --user

配置环境:

sudo gedit /etc/bash.bashrc 

将 export PATH="$PATH:$HOME/bin" 添加进去

保存:source /etc/bash.bashrc

4、编译TensorFlow

进入tensorflow文件夹中,首先进行项目配置:

./configure      # 如果只需要配置cpu环境就一直回车

注:如果选用cpu,只需要一直点确定继续就好,都是选择no;

使用bazel编译TensorFlow:

bazel build --config=opt //tensorflow:libtensorflow_cc.so

编译完成后,在bazel-bin/tensorflow中会生成两个我们需要的库文件:libtensorflow_cc.so 和 libtensorflow_framework.so

期间会因为网络了连接不上github,多尝试几次就好。

参考:Ubuntu安装TensorFlow C++_蓬莱道人的博客-CSDN博客

ubuntu18.04 tensorflow以及tensorflow lite源码编译C++库_上善若水-CSDN博客_ubuntu 编译tensorflowlite

  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要在Ubuntu上进行交叉编译TensorFlow Lite,您需要: 1. 安装交叉编译工具链:您需要安装适合您目标平台的交叉编译工具链。例如,如果您的目标平台是ARMv7,则需要安装ARMv7交叉编译工具链。 2. 下载TensorFlow Lite源代码:您可以从TensorFlow官方GitHub存储库中获取TensorFlow Lite的源代码。 3. 构建TensorFlow Lite库:在交叉编译环境中,您需要使用CMake配置TensorFlow Lite,并使用交叉编译工具链构建库。 下面是一些更详细的步骤: 1. 安装交叉编译工具链 您需要安装适合您目标平台的交叉编译工具链。例如,如果您的目标平台是ARMv7,则需要安装ARMv7交叉编译工具链。您可以使用以下命令安装ARMv7交叉编译工具链: ``` sudo apt-get install gcc-arm-linux-gnueabihf g++-arm-linux-gnueabihf ``` 2. 下载TensorFlow Lite源代码 您可以从TensorFlow官方GitHub存储库中获取TensorFlow Lite的源代码。您可以使用以下命令克隆存储库: ``` git clone https://github.com/tensorflow/tensorflow.git ``` 然后,切换到TensorFlow Lite分支: ``` cd tensorflow git checkout -b v2.5.0-rc1 tensorflow/lite ``` 3. 构建TensorFlow Lite库 在交叉编译环境中,您需要使用CMake配置TensorFlow Lite,并使用交叉编译工具链构建库。您可以使用以下命令配置CMake: ``` cd tensorflow/lite/tools/make/ ./build_host_protoc.sh make -f tensorflow/lite/tools/make/Makefile TARGET=pi3 -j 4 ``` 在上面的命令中,我们使用了pi3作为目标平台,您需要将其替换为您的目标平台。 如果一切顺利,您将在`tensorflow/lite/tools/make/gen/rpi_armv7l/bin/`目录下找到TensorFlow Lite库文件。 希望这可以帮助您成功地在Ubuntu上进行交叉编译TensorFlow Lite

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值