pytorch單機多卡訓練
訓練
只需要在model定義處增加下面一行：
model = model.to(device) # device為0號
model = torch.nn.DataParallel(model)
載入模型
- 如果是多GPU載入，沒有問題
- 如果訓練時是多GPU，但是測試時是單GPU，會出現報錯 解決辦法
pytorch單機多卡訓練
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。