Awesome
Update
- Upgrade all the PYG libraries to the newest versions.
Remove Python 3.7 support, because PyTorch 2.0 does not support it. Old update
- Add Python 3.11 and 3.12 support.
How to use
- Be sure that you have installed ROCm 5.7 or newer versions. You can use
rocm-smi
to check it.
- Go to Releases module, choose the versions of PYG you want and the right version of your Python environment.
- Download the zip file.
unzip
it, and cd
into the folder.
pip install ./*
Build environment
Ubuntu 22.04
PyTorch 2.2.2
ROCm 6.0.2
Current version
pytorch_geometric-2.5.2
pytorch_scatter-2.1.2
pytorch_sparse-0.6.18
pytorch_cluster-1.6.3
pytorch_spline_conv-1.2.2