# torch == 1.8.1
# torch-scatter
# torch-sparse
# torch-cluster
# torch-spline
# torch-geometric >= 1.4.3
# sklearn
numpy
pyyaml