pytorch正在运行:RuntimeError:预期所有张量都在同一个设备上，但至少找到了两个设备，cuda:0和cp

Question

当我运行python代码时，第44行出现runtimeError：runtimeError:预期所有张量都在同一设备上，但至少找到了两个设备，cuda:0和cpu

42  feats = self.node_features[self.train_mask]
43  labels = self.node_labels[train_mask]
44  A = torch.mm(feats.t(), feats) + 1e-05 * torch.eye(feats.size(1))
45  labels_one_hot = torch.zeros((feats.size(0), self.n_classes))

任何知道原因的人都能帮我解决吗！谢谢

Answer 1

张量torch.eye(…(似乎在CPU上。你需要通过它作为-

44  A = torch.mm(feats.t(), feats) + 1e-05 * torch.eye(feats.size(1)).to(device='cuda')

或

44  A = torch.mm(feats.t(), feats) + 1e-05 * torch.eye(feats.size(1)).cuda()