在服務器上訓練的網絡放到本地臺式機進行infer，結果出現報錯：
|
1
|
AssertionError: Invalid device id |
仔細檢查后發現原來服務器有多個GPU，當時開啟了兩個進行加速運算。
|
1
|
net1 = nn.DataParallel(net1, device_ids=[0, 1]) |
而本地臺式機只有一個GPU，調用數量超出所以報錯。
改為
|
1
|
net1 = nn.DataParallel(net1, device_ids=[0]) |
問題解決。
以上這篇解決pytorch報錯:AssertionError: Invalid device id的問題就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持服務器之家。
原文鏈接：https://blog.csdn.net/baoyongshuai1509/article/details/85254298
