编译 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