Python 3.10.13 PyTorch 1.13.0 torch_geometric 2.5.2 torch-cluster 1.6.1 torch-scatter 2.1.1 torch-sparse 0.6.17 torch-spline-conv 1.2.2 sparsemax 0.1.9 CUDA 11.7 Train RIGSL using the MELD dataset.
accelerate==0.33.0 numpy==1.24.4 ogb==1.3.6 pandas==2.2.3 PyYAML==6.0.2 rdkit_pypi==2022.9.5 scikit_learn==1.3.2 scipy==1.8.1 torch==2.1.2+cu121 torch_geometric==2.6.1 torch_scatter==2.1.2+pt21cu121 ...