在执行如下代码:
self.Model = torch.nn.DataParallel(self.Model.cuda(L.ids[0]), device_ids=L.ids)
出现这个错误:
AssertionError: Invalid device id
debud之后发现原因在于GPU只有一块卡’0’,而代码中L的id有’0’,‘1’两个值,于是把L.id改成只’0’,解决问题.
在执行如下代码:
self.Model = torch.nn.DataParallel(self.Model.cuda(L.ids[0]), device_ids=L.ids)
出现这个错误:
AssertionError: Invalid device id
debud之后发现原因在于GPU只有一块卡’0’,而代码中L的id有’0’,‘1’两个值,于是把L.id改成只’0’,解决问题.