2016-12-05 7 views

答えて

9

モデルに明示的なデバイス割り当てがない限り、これを実行できます。 IEの場合、モデルにwith tf.device('gpu:0')のようなブロックがあると、GPUなしのモデルで実行すると文句を言うでしょう。

このような場合は、あなたのインポートされたモデルは、例えば、明示的なデバイスの割り当てを持っていますが、念のためにimport_meta_graph

+0

clear_devices引数を使用していないことを確認する必要があり、それはclear_devices =真よりも、私は私のを使用できることを意味しますたとえ私のCPU上でtf.device( 'gpu:0')のようなブロックを使って訓練したとしても? –

+0

はい、デバイスをクリアし、自動配置アルゴリズムが利用可能なデバイスを使用できるようにします –

+0

[SavedModelBuilder](https: //www.tensorflow.org/api_docs/python/tf/saved_model/builder/SavedModelBuilder)クラス。 – Mark

関連する問題