pytorch-cuda=12.1
pytorch
torchvision
torchaudio
pyg
pytorch-cluster
tensorboardx
ogb
numpy=1