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

Question

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

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

我知道如何构建共享库。

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

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

Answer 1

当针对共享库构建时，我使用的标头位于$PROJECT_HOME/bazel-genfiles.

添加$PROJECT_HOME/bazel-genfiles到链接器标头列表应该就足够了。