(Windows上的TensorFlow)ImportError:无法导入名称'model_analyzer'

时间:2017-06-26 23:41:25

标签: python tensorflow virtualenv tensorflow-gpu

我正尝试使用以下步骤在Windows 10中训练TensorFlow模型:https://github.com/tensorflow/models/tree/master/attention_ocr#requirements

我已经为Windows安装了virtualenv(following this example

Windows无法识别源命令。是否需要执行virtualenv命令的附加步骤,或者是否有源命令的替代用法?

请注意,后续的pip install命令有效(pip,tensorflow_gpu),但是当我尝试使用train.py进行训练时,我收到以下错误:

 File "train.py", line 27, in <module>
    from tensorflow.contrib.tfprof import model_analyzer
ImportError: cannot import name 'model_analyzer'

1 个答案:

答案 0 :(得分:1)

看起来您的virtualenv配置正确,但您尝试使用的代码导入tf.contrib.tfprofnot currently (as of TensorFlow 1.2) supported on Windows

但是,由于tf.contrib.tfprof仅用于提供分析信息,因此您应该能够通过从train.py手动删除以下行来运行代码:

  • Line 27from tensorflow.contrib.tfprof import model_analyzer

...并在运行脚本时传递标记--show_graph_stats=false