ソースからGPUを使用するためにTensorflowをコンパイルしました。私はそれをうまくインストールすることができました。これは、サンプルmnist convolutional.py
スクリプトを正常に実行することができます。Tensorflow:ImportErrorはretrain.pyの名前graph_utilをインポートできません
私はtensorflow/examples/image_retraining
でretrain.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
python 2.7のソースを実行している場合、graph_util.pyのインポートパッケージをtensorfloy.python.clientファイルに手動でretlin.pyファイルに変更する必要があるようです – pleonasmik