标签 > module must have parameters on cuda:0 but found cuda:2：DataParallel 主卡与模型位置要一致

这个报错出自 torch nn DataParallel，根源是 DataParallel 默认主卡是 cuda:0，但是你的模型参数被提前放到了 cuda:2 上，主设备和