如何使用bazel编译tensorflow链接已编译的grpc库?

时间:2017-09-22 14:03:38

标签: tensorflow bazel tensorflow-gpu

我从github repo克隆了tensorflow(版本1.2)。我在分布式tensorflow中遇到grpc的问题。基本上我想用最新的grpc库编译tensorflow(1.2)。

我下载并编译了grpc库,所以我有库文件和标题。在构建张量流时,如何指示bazel采用已编译的库文件和头文件。

1 个答案:

答案 0 :(得分:2)

没有简单的方法。如果我这样做,我会克隆tensorflow,并更新tensorflow/workspace.bzl以提取更新的grpc版本,或将patched_http_archive替换为new_local_repositorynew_http_archive。我假设您需要应用third_party/grpc中的补丁。然后,您应该能够使用默认过程构建tensorflow:./configurebazel build --config=opt //tensorflow/tools/pip_package:build_pip_package

快速扫描tensorflow BUILD文件似乎tensorflow使用grpc的third_party依赖(例如nanopb),因此无论如何都不会有编译库和头文件。如果您坚持使用预构建的grpc,则需要提供这些依赖项,并更新BUILD文件和WORKSPACE文件。 Bazel没有这个功能'用其内置的输出替换存储库'。