Importing pytorch geometric results in an error message...

Pytorch Geometric Expected all tensors to be on the same device [demo code not working]...