7
私は、GPUで訓練されたモデルがCPUとの推論を実行するために使用できるかどうか疑問に思っていましたか? (そしてその逆) ありがとう!gpuで訓練されたモデルを推論にcpuで使用できますか?
私は、GPUで訓練されたモデルがCPUとの推論を実行するために使用できるかどうか疑問に思っていましたか? (そしてその逆) ありがとう!gpuで訓練されたモデルを推論にcpuで使用できますか?
モデルに明示的なデバイス割り当てがない限り、これを実行できます。 IEの場合、モデルにwith tf.device('gpu:0')
のようなブロックがあると、GPUなしのモデルで実行すると文句を言うでしょう。
このような場合は、あなたのインポートされたモデルは、例えば、明示的なデバイスの割り当てを持っていますが、念のためにimport_meta_graph
で
clear_devices
引数を使用していないことを確認する必要があり、それはclear_devices =真よりも、私は私のを使用できることを意味しますたとえ私のCPU上でtf.device( 'gpu:0')のようなブロックを使って訓練したとしても? –はい、デバイスをクリアし、自動配置アルゴリズムが利用可能なデバイスを使用できるようにします –
[SavedModelBuilder](https: //www.tensorflow.org/api_docs/python/tf/saved_model/builder/SavedModelBuilder)クラス。 – Mark