【已解决】RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cu

问题描述

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1!

解决办法 

        这个办法比较常见，就是说在训练的时候需要张量在同一个设备上，但是呢不小心放到了其他的设备，所以有这个错误，那怎么办呢？调呗

device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')

改为

device = torch.device('cuda:1' if torch.cuda.is_available() else 'cpu')

        再ctr+F在各个py文件查找关键词“cuda” ，找到了

        改！！！ 

device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')

改为

device = torch.device('cuda:1' if torch.cuda.is_available() else 'cpu')

        齐活！

总结

        其实写到这里会发现上面两部分代码是一样的，无所谓，主要是证明两个得是一样的，仅此而已。