报错:Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1!
解决方法:将train模块里面一些比如model.cuda(),image.cuda(),mask.cuda()之类的,改为model.to(device),image.to(device),mask.to(device)。
当然device在开头的时候要定义一下device = torch.device('cuda:1')
我猜应该是.cuda默认就是cuda:0