使用 Bazel 运行时,我应该在哪里保存 Tensorflow 的 .pb 图?



目录结构:

~/tensorflow/tensorflow/cc/dnnops/
├── BUILD
├── graph.pb
└── main.cc

来自main.cc的失败代码行:status = ReadBinaryProto(tf::Env::Default(), "graph.pb", &graph_def);。完整的代码可以在这里找到。

巴泽尔输出:

警告:/home//.cache/bazel/_bazel_rd1/4ab077b6e1a2076b6ea9f23b417088a6/external/protobuf_archive/WORKSPACE:1:/home//.cache/bazel/bazel/4ab077b6e1a2076b6ea9f23b417088a6/external/protobuf_archive/WORKSPACE (@com_google_protobuf( 中给出的名称与存储库定义 (@protobuf_archive( 中给出的名称不匹配;这将导致未来版本中的构建错误 信息:分析目标//tensorflow/cc/dnnops:dnnops (0 包已加载(。 信息:找到 1 个目标... Target//tensorflow/cc/dnnops:dnnops up-to-date: Bazel-bin/tensorflow/cc/dnnops/dnnops 信息:运行时间:5.378秒,关键路径:5.04秒 信息:2个进程,本地。 信息:构建成功完成,总共 3 个操作

信息:运行命令行:bazel-bin/tensorflow/cc/dnnops/dnnops 未找到: 图形.pb;没有这样的文件或目录 错误:命令中的非零返回代码"1":进程退出,状态为 1

我尝试了什么

bazel-bin/tensorflow/cc/dnnops中复制相同的图形文件。仍然失败,Bazel 的输出相同。

问题

我应该如何将图形文件位置公开给Tensorflow/Bazel?

愚蠢的疏忽...我输入了完整的路径。由此:

status = ReadBinaryProto(tf::Env::Default(), "graph.pb", &graph_def);

对此:

status = ReadBinaryProto(tf::Env::Default(), "/home/<user>/path/to/graph.pb", &graph_def);

最新更新