RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1!
将
model = CNN().to(device) print(summary(model, (1, wn, ws, 6)))
改为
print(summary(CNN().to('cuda'), (1, wn, ws, 6), device='cuda'))
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1!
将
model = CNN().to(device) print(summary(model, (1, wn, ws, 6)))
改为
print(summary(CNN().to('cuda'), (1, wn, ws, 6), device='cuda'))