Ubuntu 16.04 build TensorFlow & Lite C++ 动态连接库

最新推荐文章于 2025-11-06 09:18:24 发布

原创

最新推荐文章于 2025-11-06 09:18:24 发布 · 3k 阅读

1 ·

CC 4.0 BY-SA版权

本文指导如何在Ubuntu 16.04上构建TensorFlow的C++动态链接库以及TensorFlow Lite。首先安装bazel工具，然后通过git克隆TensorFlow源码。使用bazel构建TensorFlow，可以添加Intel CPU加速选项。对于TensorFlow Lite，需要编辑BUILD文件并构建libtflite.so。最后，提供了如何构建TensorFlow的pip安装包的步骤。

首先要安装bazel tool

下载源码：git clone https://github.com/tensorflow/tensorflow.git

Build Tensorflow:

bazel build --jobs 6 //tensorflow:libtensorflow_cc.so

假如需要增加Intel CPU 加速:

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

Build Tensorflow Lite:

编辑TensorFlow source code里面的"tensorflow/contrib/lite/BUILD"文件，增加：

cc_binary(
name = "libtflite.so",
deps = [":framework",
"//tensorflow/contrib/lite/kernels:builtin_ops",
"//tensorflow/contrib/lite/kernels:eigen_support",
"//tensorflow/contrib/lite/kernels:gemm_support",
],
linkopts=["-shared -Wl,--whole-archive" ],
linkshared=1

)

bazel build --jobs 6 //tensorflow/contrib/lite:libtflite.so --cxxopt="-std=c++11" --copt="-O3"