2017-03-22 17 views
-3

モデルはGPUでトレーニングされ、結果はチェックポイントファイルによって保存されます。保存されたチェックポイントファイルはcpu-tensorflowで実行できますか?そうでない場合、cpu-tensorflowでモデルを実行するために保存されたチェックポイントファイルを変換できますか?Tensorflow:GPUで訓練されたモデル(チェックポイントファイル)をモデル実行中のCPUに変換できますか?

+7

[推論のためにCPU上でgpuで訓練されたモデルを使用できますか?](http://stackoverflow.com/questions/40980035/can-a-model-trained-on-gpu-used- on-cpu-for-forferenceおよびその逆) – kaufmanu

答えて

0

はい!それは通常できる!

例外はwith tf.device('gpu:0')です。あなたがあなたのコードにそれらを持っていないなら、あなたは行くのが良いです!

幸運を祈る!

+1

コードにデバイス割り当てがある場合、remove_device書き換えルールを使用して、保存されたGraphDef .pbファイルからそれらを削除できます。https://github.com/テンソルフロー/テンソルフロー/ツリー/マスタ/テンソルフロー/ tools/graph_transforms /#remove_device –

関連する問題