减少libtensorflow.so文件

Question

我正在尝试将具有C ++接口的Tensorflow库集成到我的C ++应用程序中。问题是使用

直接构建

bazel build (some options) //tensorflow:libtensorflow.so

制作一个168Mb的libtensorflow.so文件。这对于我的应用程序来说太多了。我找到了一些有关减少Android库大小的指南，但找不到针对常规桌面构建目标的任何指南。

我认为libtensorflow.so具有TF的所有功能，但我真正需要的是具有基本Conv操作等的推理引擎。

有什么建议吗？

谢谢！

Answer 1

您可能想尝试CMake build。它为您的案例提供了两个有趣的构建选项：

• 设置$sub不会从tensorflow_BUILD_CONTRIB_KERNELS=OFF构建内核。
• 设置tf.contrib仅构建最常见内核的一小部分。