模型使用GPU訓練,結果由檢查點文件保存。保存的檢查點文件可以通過cpu-tensorflow運行?如果沒有,可以轉換已保存的檢查點文件以在cpu-tensorflow中運行模型?Tensorflow:GPU上的訓練模型(檢查點文件)可以轉換爲CPU運行模型?
-3
A
回答
0
是的!它通常可以!
例外是with tf.device('gpu:0')
陳述。如果你的代碼中沒有他們,那麼你很好!
祝你好運!
+1
如果您的代碼中有設備分配,可以使用remove_device重寫規則將它們從保存的GraphDef .pb文件中刪除:https://github.com/ tensorflow/tensorflow /樹/主/ tensorflow /工具/ graph_transforms /#remove_device –
相關問題
- 1. 我可以用cpu訓練初始v3模型嗎?
- 2. NLP模型訓練
- 3. TensorFlow:是否有可能爲多GPU訓練恢復檢查點模型?
- 4. 是否可以使用GPU訓練H2O模型並使用CPU進行預測?
- 5. 重新訓練tensorflow模型
- 6. 如何訓練ML模型?
- 7. 訓練分類模型Opennlp
- 8. Keras訓練稀疏模型
- 9. 錯誤訓練im2txt模型
- 10. LightGBM:繼續訓練模型
- 11. Tensorflow:使用預訓練以來模型
- 12. 在openai cartole上訓練tensorflow模型
- 13. 並行訓練多個Auto.Arima模型
- 14. 與sklearn平行訓練多個模型?
- 15. 可以在cpu上使用gpu訓練模型以進行推理,反之亦然?
- 16. 使用CPU vs GPU來訓練模型 - 速度vs內存
- 17. 加載訓練有素的Keras模型並繼續訓練
- 18. 文字處理 - 短語檢測後的Word2Vec訓練(bigram模型)
- 19. 訓練方法seq2seq模型的選擇
- 20. 還原訓練的模型在Tensorflow 1.2
- 21. dlib dnn_face_recognition_ex訓練模型沒有檢測到訓練圖像的人
- 22. 哪裏可以下載MXNet預訓練的模型文件最終0000.params
- 23. 在keras預訓練模型中,我可以刪除圖層嗎?
- 24. 加載訓練好的模型並在節點js中運行測試
- 25. 分割訓練數據以訓練n個模型的最佳數量
- 26. 意大利語Syntaxnet預訓練模型
- 27. CNTK python api - 繼續訓練模型
- 28. Tensorflow:它如何訓練模型?
- 29. Tensorflow重新訓練模型性能
- 30. Tensorflow:在C++中訓練模型
可能的重複[可以在cpu上使用CPU進行推理的模型,反之亦然?](http://stackoverflow.com/questions/40980035/can-a-model-trained-on-gpu-used- cpu-for-inference-and-反之亦然) – kaufmanu