python - ImportError：无法导入名称 rewriter_config_pb2

Question

我只是在遵循 tensorflow 的教程并执行以下 cmd ：

python ptb_word_lm.py --data_path=/home/priyankit/data/validsrc2-hi --
model=small

它显示以下错误：

Traceback (most recent call last):
File "ptb_word_lm.py", line 68, in 
import util
File "/home/priyankit/models-master/tutorials/rnn/ptb/util.py", line 
23, in 
from tensorflow.core.protobuf import rewriter_config_pb2
ImportError: cannot import name rewriter_config_pb2

链接到存储库

Answer 1

你需要做两件事。首先，通过执行以下操作升级 tensorflow：

pip install --upgrade tensorflow

其次，如果您不使用 GPU，则使用 0 GPU 执行脚本，如下所示：

python ptb_word_lm.py --data_path=/home/priyankit/data/validsrc2-hi --model=small --num_gpus=0

Answer 2

当我尝试在他们的网站上做 RNN Tensorflow 教程时，我遇到了完全相同的错误循环。

在终端上运行我的程序时，对我有用（我没有 GPU）的方法是在“--num_gpus=0”末尾包含这个。