title: pytorch running: RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu
body: <p>When I running the python code, there is a runtimeError in line 44:<strong>RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!</strong></p>
  <pre><code>    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))
  </code></pre>
  <p>Can anyone who knows the reason and help me fix it!  Thanks!</p>