使用torchsummary时出现错误RuntimeError: Expected all tensors to be on the same device, ...

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'))