在 tensorflow 服务中静态链接来自 .a 文件的自定义操作

Question

我有一个自定义操作实现了 CUDA，并使用像 hdrnet 这样的 Makefile 构建。我可以构建 .so 并在 tensorflow 中导入。对于 tf-serving，需要静态链接 .a 文件，但所有教程都参考了 custom op 的 bazel 构建过程，而不是直接从 .a 文件链接已编译的操作。 我是否必须编写示例引用的构建过程，或者我可以直接使用 .so/.a 文件构建 tf-serving？

Answer 1

我最终使用 op-linked in 从源代码编译 tensorflow-serving。Tensorflow 教程对此并不完整，并且缺少其他依赖项，我在此问题 issue 中解决了这些依赖项。