NEEDED:
torch                     1.4.0
torch-geometric    1.5.0
torch-cluster         1.5.2
torch-scatter         2.0.3
torch-sparse          0.6.0
torch-spline-conv  1.2.0