首页 文章

使用bazel时未发现Tensorflow toco pacakage

提问于
浏览
1

我试图使用以下命令将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 回答

  • 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
    

相关问题