gpt4 book ai didi

python - 嵌套的 Bazel 项目

转载 作者:太空狗 更新时间:2023-10-30 01:20:41 27 4
gpt4 key购买 nike

我正在尝试构建一个使用 TensorFlow Serving 的项目,所以我创建了一个包含 WORKSPACE 文件的目录 my_dir,将服务存储库克隆到其中,将我的自定义文件放入目录 my_project,在 tensorflow_serving 中配置 tensorflow,构建 tensorflow从 my_dir 服务/服务

bazel build //tensorflow_serving/...

那里的一切都很好,然后我尝试构建一个模仿 mnist_export 的 python 文件并将其放入 my_dir 并制作一个 BUILD 文件

py_binary(
name = "export_cnn",
srcs = [
"export_cnn.py",
],
deps = [
"@tf//tensorflow:tensorflow_py",
"@tf_serving//tensorflow_serving/session_bundle:exporter",
],
)

但是,当我运行

bazel build //my_project:export_cnn

我收到以下错误:

ERROR: 

.../bazel/_bazel_me/3ef3308a843af155635e839105e8da5c/external/tf/tensorflow/core/BUILD:92:1: null failed: protoc failed: error executing command bazel-out/host/bin/external/tf/google/protobuf/protoc '--cpp_out=bazel-out/local_linux-fastbuild/genfiles/external/tf' -Iexternal/tf -Iexternal/tf/google/protobuf/src ... (remaining 1 argument(s) skipped).
tensorflow/core/framework/step_stats.proto: File not found.
tensorflow/core/framework/device_attributes.proto: File not found.
tensorflow/core/framework/graph.proto: File not found.
tensorflow/core/framework/tensor.proto: File not found.
tensorflow/core/protobuf/config.proto: File not found.
tensorflow/core/protobuf/worker.proto: Import "tensorflow/core/framework/step_stats.proto" was not found or had errors.
tensorflow/core/protobuf/worker.proto: Import "tensorflow/core/framework/device_attributes.proto" was not found or had errors.
tensorflow/core/protobuf/worker.proto: Import "tensorflow/core/framework/graph.proto" was not found or had errors.
tensorflow/core/protobuf/worker.proto: Import "tensorflow/core/framework/tensor.proto" was not found or had errors.
tensorflow/core/protobuf/worker.proto: Import "tensorflow/core/protobuf/config.proto" was not found or had errors.
tensorflow/core/protobuf/worker.proto:41:12: "DeviceAttributes" is not defined.
tensorflow/core/protobuf/worker.proto:64:3: "GraphDef" is not defined.
tensorflow/core/protobuf/worker.proto:72:3: "GraphOptions" is not defined.
tensorflow/core/protobuf/worker.proto:141:3: "TensorProto" is not defined.
tensorflow/core/protobuf/worker.proto:180:3: "StepStats" is not defined.
tensorflow/core/protobuf/worker.proto:225:3: "BusAdjacency" is not defined.
tensorflow/core/protobuf/worker.proto:227:3: "BusAdjacency" is not defined.
tensorflow/core/protobuf/worker.proto:232:3: "TensorProto" is not defined.
tensorflow/core/protobuf/worker.proto:272:3: "StepStats" is not defined.

在我的 WORKSPACE 文件中,我有以下内容:

local_repository(
name = "tf",
path = __workspace_dir__ + "/serving/tensorflow",
)

local_repository(
name = "tf_serving",
path = __workspace_dir__ + "/serving",
)

load('//serving/tensorflow/tensorflow:workspace.bzl', 'tf_workspace')
tf_workspace("serving/tensorflow/", "@tf")

我的假设是,因为 tensorflow 是一个子子项目,所以它不会将其生成的文件放在祖父项目 bazel-out 中。但是,我已经尝试了很多东西,但无法让它发挥作用。

最佳答案

我在一个文件夹中提供 tensorflow,在另一个文件夹中提供我的项目。这是我的工作区文件:

workspace(name = "my_project")

local_repository(
name = "org_tensorflow",
path = __workspace_dir__ + "/tf-serving/tensorflow/",
)

local_repository(
name = "tf_serving",
path = __workspace_dir__ + "/tf-serving/",
)

load('//tf-serving/tensorflow/tensorflow:workspace.bzl', 'tf_workspace')
tf_workspace("tf-serving/tensorflow/", "@org_tensorflow")

# ===== gRPC dependencies =====

bind(
name = "libssl",
actual = "@boringssl_git//:ssl",
)

bind(
name = "zlib",
actual = "@zlib_archive//:zlib",
)

我还将 zlib.BUILD 从 tensorflow 服务复制到我拥有 WORKSPACE 文件的同一位置。

我项目中的 BUILD 文件有这个规则(和你的类似):

py_binary(
name = "export_model",
srcs = [
"export_model.py",
],
deps = [
"@org_tensorflow//tensorflow:tensorflow_py",
"@tf_serving//tensorflow_serving/session_bundle:exporter",
],
)

我的代码和你的代码之间的区别在于我将依赖项包含在我的根 WORKSPACE 中。这段代码在一台机器上对我来说编译和工作正常我有一些问题在其他机器上编译它(ubuntu 14.04)因为一个依赖。我希望它对你有用。

关于python - 嵌套的 Bazel 项目,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37581247/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com