Solved by ptrblck in post #2 
              
              
                Maybe your torch_geometric version is too old. Note that this library is not shipped in the PyTorch installation, but is installed separately, so you would have to check its version.