代码之家  ›  专栏  ›  技术社区  ›  Hossein

获取错误:尝试在tensorflow中训练mobilenet时,任何.rc文件中都未定义配置值cuda

  •  3
  • Hossein  · 技术社区  · 6 年前

    我想逃跑 MobileNet_v1 ImageNet 为此我用 official Tensorflow Model repository 跟随他们的指引。
    但是当我真的试着为 MobileNet_v1

    models/research/slim$ bazel build -c opt --config=cuda mobilenet_v1_{eval,train}
    

    我得到:

    Starting local Bazel server and connecting to it...
    INFO: Options provided by the client:
      Inherited 'common' options: --isatty=1 --terminal_columns=146
    ERROR: Config value cuda is not defined in any .rc file
    

    之前的所有步骤都很成功,除了这一步,一切看起来都很好。
    我用的是:

    Ubuntu 16.04  
    TF version: v1.7.0 (and 1.10.1) 
    Cuda v9.0  
    cuDNN v7.0 (and 7.1.3)  
    bazel release 0.16.1
    
    1 回复  |  直到 6 年前
        1
  •  5
  •   Ben Price    6 年前

    根据 bazel releases page ,

    configs是一个错误。

    从0.16.0及以上版本开始。