c++ - 有没有办法为 C++ api 打包 Tensorflow？

Question

我一直在使用 Tensorflow c++ api 开发 c++ 项目。它只是执行从 Python 创建的 tensorflow 图。我现在使用带有 Tensorflow 代码的 bazel 构建它。但我认为这是低效的方式。

我只想要 Tensorflow 库和头文件，并且只使用 Cmake 编译我的项目。

我知道如何构建共享库。

bazel build -c opt --config=cuda//tensorflow:libtensorflow.so 但是这个命令只是生成一个 libtensorflow.so 文件。我找不到用于构建我的项目的头文件。

有没有办法为 C++ 打包 tensorflow 库？例如mvn package命令。

Answer 1

据我所知，没有官方可分发的 C++ API 包。然而，有 tensorflow_cc为您构建和安装 TF C++ API 的项目，以及您可以链接的方便的 CMake 目标。根据您的描述，这可能正是您所需要的。