c++ - 将预编译的tensorflow与cmake一起使用

Question

我在 CLion 中设置了一个使用 CMake 的 C++ 项目。我正在使用各种第 3 方库，并且还想集成 Tensorflow。

我试过 bazel 将 Tensorflow 编译成一个共享库 libtensorflow.so 哪种方法可行，但是仍然有很多我必须修复的依赖项(例如，当前的 protobuf 版本，一旦我这样做了，就会有更多)。

有没有办法使用标准的 Tensorflow git 存储库并以某种方式链接为 python 使用而预编译的库？或者有其他方便的方法吗？

Python 中的 Tensorflow 非常适合我。

Answer 1

我知道这个答案已经很晚了，但我遇到了你的确切问题并且能够解决它。我创建了一个存储库 here描述了如何准确地完成你想要的。要点是:

• 将构建规则添加到包含所有必需的 C++ 元素的 TensorFlow 存储库。
• 使用 Bazel 构建共享库并将所有 header 复制到 /usr/local。
• 安装特定版本的 Protobuf 和 Eigen(这是通过脚本自动完成的)或将它们添加为外部依赖项。
• 使用给定的文件配置您的 CMake 项目。

如果您有任何疑问或问题，请随时与我联系。