无法使用bazel构建构建TensorFlow模型

时间:2016-10-24 19:58:40

标签: tensorflow bazel tensorflow-serving

我已按照https://tensorflow.github.io/serving/serving_basic中的说明在Windows机器上运行的docker机器中设置了TensorFlow服务器。我成功地能够构建并运行mnist_model。但是,当我尝试通过运行以下命令“bazel build //tensorflow_serving/example:wide_n_deep_tutorial.py”为wide_n_deep_tutorial示例构建模型时,模型未成功构建,因为bazel-bin文件夹中没有生成文件。

enter image description here

由于在构建模型时没有显示错误消息,因此我无法找出问题所在。如果有人可以帮助我调试并解决问题,我将非常感激。

2 个答案:

答案 0 :(得分:0)

您只是在这里猜测命令行为there is no target in the BUILD file of tensorflow serving for wide_n_deep_tutorial.py

截至今天,您只能构建mnist和inception目标。

答案 1 :(得分:0)

通过在BUILD文件中添加宽和深模型的目标来解决问题。

在BUILD文件中添加了以下内容:

py_binary(
    name = "wide_n_deep_model",
    srcs = [
        "wide_n_deep_model.py",
    ],
    deps = [
        "//tensorflow_serving/apis:predict_proto_py_pb2",
        "//tensorflow_serving/apis:prediction_service_proto_py_pb2",
        "@org_tensorflow//tensorflow:tensorflow_py",
    ],
)