如何使用 TensorFlow C++ api 作为共享库与 bazel？

Question

Now I can build TensorFlow C++ api as a shared library refer to make shared libraries with Bazel at Tensorflow , libtensorflow_cc.so and libtensorflow_framework.so file can be locate at bazel-bin/tensorflow .

我对bazel不熟悉，我应该如何使用那些带有bazel的TF共享库来编写C++代码，你能提供一个这样的例子吗？

Answer 1

Bazel只是一个类似于 make 或 cmake 的工具。 要使用 bazel 编译代码，您必须创建一个构建文件BUILD 。 假设您的代码位于例如tensorflow/tensorflow/test/code.cpp ：

#include "tensorflow/core/public/session.h"
#include "tensorflow/core/platform/env.h"

using namespace tensorflow;

...

创建一个包含如下内容的BUILD文件（在同一文件夹中）：

cc_binary(
    name = "code",
    srcs = ["code.cc"],
    deps = [
        "//tensorflow/core:tensorflow",
    ]
)

然后使用命令配置和编译您的代码：

cd tensorflow
./configure
cd tensorflow/tensorflow/test
bazel build :code

可执行文件将在 tensorflow 根目录中的bazel-bin文件夹中可用。

---

但是，编译后的代码可能很大（> 100 MB），因此最好将您的代码链接到 TensorFlow 的共享库。 libtensorflow_cc.so和libtensorflow_framework.so都可以像其他共享库文件 ( .so ) 一样在您的代码中使用。 以下是执行此操作的常用方法。

1. 使用共享库（ libtensorflow_cc.so或/和libtensorflow_framework.so ），使用 header 库文件 ( .h ) 编译 C++ 代码，例如，

    g++ -fPIG -Wall -o code code.cpp -L/usr/lib/libtensorflow_cc.so -ltensorflow_cc

2. 将LD_LIBRARY_PATH设置为.so文件的位置
3. 使用ldd your_code.out来确认您的可执行文件是否正确链接到共享库。
4. 运行可执行文件（例如./your_code.out ）