2016-06-16 4 views
1

ソースからGPUを使用するためにTensorflowをコンパイルしました。私はそれをうまくインストールすることができました。これは、サンプルmnist convolutional.pyスクリプトを正常に実行することができます。Tensorflow:ImportErrorはretrain.pyの名前graph_utilをインポートできません

私はtensorflow/examples/image_retrainingretrain.pyスクリプトを実行するときしかし、それは次のようなエラー吐く:コマンド及びその他の出力で

Traceback (most recent call last): 
    File "retrain.py", line 70, in <module> 
    from tensorflow.python.client import graph_util 
ImportError: cannot import name graph_util 

を:

$ python retrain.py \ 
--bottleneck_dir=gpu_train/bottlenecks \ 
--model_dir=inception \ 
--output_graph=gpu_train/retrained_graph.pb \ 
--output_labels=gpu_train/retrained_labels.txt \ 
--image_dir categories 
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcublas.dylib locally 
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcudnn.dylib locally 
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcufft.dylib locally 
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcuda.dylib locally 
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcurand.dylib locally 
Traceback (most recent call last): 
    File "retrain.py", line 70, in <module> 
    from tensorflow.python.client import graph_util 
ImportError: cannot import name graph_util 

答えて

4

移動commit昨日がありましたファイルgraph_util.pyからtensorflow.python.clientからtensorflow.python.frameworkまでです。

のの旧バージョンを持っていますが、graph_util.pyも移動してしまったのは間違いです。


とにかく、graph_util.pymaster versionは(from tensorflow.python.framework import graph_utilで)最新であるので、あなたは、最新バージョン(単に引っ張る、再びコンパイルする必要はありません)を取得する必要があります。

+0

python 2.7のソースを実行している場合、graph_util.pyのインポートパッケージをtensorfloy.python.clientファイルに手動でretlin.pyファイルに変更する必要があるようです – pleonasmik

関連する問題