numpy~=1.21.2
pandas~=1.2.3
scikit-learn~=1.0.2
networkx~=2.6.2
node2vec~=0.4.4
scipy~=1.7.3
nni~=2.4

[cpu]
torch==1.8.1
torchvision==0.9.1
torchaudio==0.8.1

[gpu]
torch==1.8.1+cu101
torchvision==0.9.1+cu101
torchaudio==0.8.1
