AssertionError: Invalid device id

在执行如下代码:

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’,解决问题.

你可能感兴趣的:(一堆bug,pytorch)