移動グラフは、CPU上でテストされるGPUを訓練しました。

Question

私はGPUでトレーニングするこのCNNを持っています。トレーニング中に、私は定期的にチェックポイントを保存します。

その後、.metaファイルとチェックポイントを読み取る小さなスクリプトを用意し、CPUでいくつかのテストを行います。私は、次のコードを使用します。私はセーバーがGPU上での操作を置くしようとしていることを教えて、このエラーを得続ける

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) 
with sess.as_default(): 
    with tf.device('/cpu:0'): 
     saver = tf.train.import_meta_graph('{}.meta'.format(model)) 
     saver.restore(sess,model) 

。

どうすれば変更できますか？

Answer 1

ハック回避策、あなたはそれが定期的セーバーを使用していますが、輸出はclear_devicesでこれを行う方法をTensorFlow見ることができます（.pbtxtで終わる）あなたのグラフ定義ファイルを開いて、プログラム的なアプローチについてdevice:

で始まるすべての行を削除し、メタグラフではない輸出者