使用bazel时未发现Tensorflow toco pacakage

时间:2018-05-11 23:40:11

标签: tensorflow bazel tensorflow-lite

我正在尝试使用以下命令将pb文件转换为tlite文件:

bazel run // tensorflow / contrib / lite / toco:toco --input_file = sample.pb --output_file = sample.tlite

我收到以下错误: 错误:正在跳过// tensorflow / contrib / lite / toco:toco:在程序包路径上找不到构建文件。

我在Python文件夹下安装了tensorflow和toco。 Bazel run命令的正确途径是什么?

1 个答案:

答案 0 :(得分:2)

在python中安装toco不起作用。 您必须在tensorflow源目录中执行此命令。 有关依赖关系和配置的必要说明,请参阅https://www.tensorflow.org/install/install_sources

另请注意,在bazel args和命令args之间需要一个“ - ”:

bazel run //tensorflow/contrib/lite/toco:toco -- --input_file=sample.pb --output_file=sample.tlite