Tensorflow：GPU上的訓練模型（檢查點文件）可以轉換爲CPU運行模型？

-3

模型使用GPU訓練，結果由檢查點文件保存。保存的檢查點文件可以通過cpu-tensorflow運行？如果沒有，可以轉換已保存的檢查點文件以在cpu-tensorflow中運行模型？Tensorflow：GPU上的訓練模型（檢查點文件）可以轉換爲CPU運行模型？

來源

2017-03-22 liu bob

可能的重複[可以在cpu上使用CPU進行推理的模型，反之亦然？]（http://stackoverflow.com/questions/40980035/can-a-model-trained-on-gpu-used- cpu-for-inference-and-反之亦然） – kaufmanu

是的！它通常可以！

例外是with tf.device('gpu:0')陳述。如果你的代碼中沒有他們，那麼你很好！

祝你好運！

來源

2017-03-22 13:47:42 rmeertens

如果您的代碼中有設備分配，可以使用remove_device重寫規則將它們從保存的GraphDef .pb文件中刪除：https：//github.com/ tensorflow/tensorflow /樹/主/ tensorflow /工具/ graph_transforms /＃remove_device –

Tensorflow：GPU上的訓練模型（檢查點文件）可以轉換爲CPU運行模型？

回答

相關問題