python - （Windows 上的 TensorFlow）ImportError: cannot import name 'model_analyzer'

Question

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

我已经为 Windows 安装了 virtualenv（按照这个例子）

Windows 无法识别 source 命令。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'

Answer 1

看起来您的 virtualenv 配置正确，但是您尝试使用的代码 imports tf.contrib.tfprof，目前（从 TensorFlow 1.2 开始）在 Windows 上不受支持。

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

• 第 27 行：from tensorflow.contrib.tfprof import model_analyzer

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