Skip to content

Looong01/pyg-rocm-build

Repository files navigation

Update

  1. Upgrade all the PYG libraries to the latest versions.
  2. Remove Python 3.9 support, because PyTorch 2.9 does not support it.
  3. Add Python 3.14 support.
  4. Support ROCm 7.0.2.

How to use

  1. You can find all the pre-build packages here. Please follow the tags and find which version of ROCm and PyTorch you do need!
  2. Be sure that you have installed ROCm 5.7 or newer versions. You can use rocm-smi to check it.
  3. Go to Releases module, choose the versions of PYG you want and the right version of your Python environment.
  4. Download the zip file.
  5. unzip it, and cd into the folder.
  6. pip install ./*

Attention

  1. The Python 3.8 version of these wheels was built by PyTorch 2.5 building with ROCm 6.1, because ROCm-6.2-build PyTorch doesn't support Python 3.8 in the future.
  2. The Python 3.9 version of these wheels was built by PyTorch 2.8 building with ROCm 6.4, because ROCm-7.0-build PyTorch doesn't support Python 3.9 in the future.

Build environment

Ubuntu 24.04 LTS
PyTorch 2.7 for Python 3.10~3.14
ROCm 7.0.2
GNU-13
Clang-20

Current version

pytorch_geometric-2.7.0
pytorch_scatter-2.1.2
pytorch_sparse-0.6.18
pytorch_cluster-1.6.3
pytorch_spline_conv-1.2.2

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Packages

No packages published