torch
torch_geometric>=1.7.0
