编译 tensorflowlite_c.so(Linux)

info
环境

Debian 12, Linux Kernel 6.1.0

一、安装需要的基础软件

apt install -y git curl gcc g++

二、下载 Bazelisk

# 从 Github 下载最新版的 bazelisk-linux-amd64,命名为 bazelisk
curl -L https://github.com/bazelbuild/bazelisk/releases/latest/download/bazelisk-linux-amd64 -o bazelisk

# 授予运行权限
chmod +x ./bazelisk

# 将 bazelisk 移动到 /usr/local/bin/
sudo mv ./bazelisk /usr/local/bin/

三、下载 TensorFlow 源码

git clone https://github.com/tensorflow/tensorflow.git tensorflow_src
cd tensorflow_src

# 使用 git checkout <版本号> 来选择需要的 TensorFlow 版本,如:
git checkout v2.15.0

四、开始编译

bazelisk build -c opt //tensorflow/lite/c:libtensorflowlite_c.so

五、编译结果

success
success

INFO: Analyzed target //tensorflow/lite/c:libtensorflowlite_c.so (0 packages loaded, 0 targets configured). INFO: Found 1 target... Target //tensorflow/lite/c:libtensorflowlite_c.so up-to-date: bazel-bin/tensorflow/lite/c/libtensorflowlite_c.so INFO: Elasped time: 108.942s, Critical Path: 35.73s INFO: 773 processes: 147 internal, 626 local. INFO: Build completed successfully, 733 total actions

大功告成!

构建成功后,动态链接库位于:

.\tensorflow_src\bazel-bin\tensorflow\lite\c\tensorflowlite_c.so

参考

Build LiteRT for ARM boards  |  Google AI Edge  |  Google AI for Developers