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

时间:2018-06-01 09:03:06

标签: tensorflow cpu x86-64 bazel

目录结构:

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

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

Bazel输出:

  

警告:/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个目标......   目标// tensorflow / cc / dnnops:dnnops最新:     巴泽勒滨/ tensorflow /立方厘米/ dnnops / dnnops   信息:经过的时间:5.378s,关键路径:5.04s   信息:2个进程,本地。   信息:构建成功完成,总共3个动作

     

信息:运行命令行:bazel-bin / tensorflow / cc / dnnops / dnnops   找不到:graph.pb;没有相应的文件和目录   错误:非归零码' 1' from命令:进程退出状态1

我尝试了什么

bazel-bin/tensorflow/cc/dnnops内复制相同的图表文件。仍然无法使用Bazel的相同输出。

问题

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

1 个答案:

答案 0 :(得分:1)

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

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

对此:

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