From 9d88a0b6f46e3055ed5a8a65169fb23b2910c30f Mon Sep 17 00:00:00 2001 From: Matthias Fey Date: Tue, 12 Nov 2024 09:23:58 +0100 Subject: [PATCH] PyTorch 2.5 support (#232) * update * update --- .github/workflows/building-conda.yml | 4 +++- .github/workflows/building.yml | 4 +++- .github/workflows/testing.yml | 6 +++--- README.md | 27 +++++++++++++-------------- conda/pytorch-cluster/README.md | 2 +- 5 files changed, 23 insertions(+), 20 deletions(-) diff --git a/.github/workflows/building-conda.yml b/.github/workflows/building-conda.yml index 44b115d..f02a18a 100644 --- a/.github/workflows/building-conda.yml +++ b/.github/workflows/building-conda.yml @@ -12,9 +12,11 @@ jobs: matrix: os: [ubuntu-20.04, macos-14, windows-2019] python-version: ['3.8', '3.9', '3.10', '3.11', '3.12'] - torch-version: [2.4.0] # [2.3.0, 2.4.0] + torch-version: [2.5.0] # [2.3.0, 2.4.0, 2.5.0] cuda-version: ['cpu', 'cu118', 'cu121', 'cu124'] exclude: + - torch-version: 2.5.0 + python-version: '3.8' - torch-version: 2.3.0 cuda-version: 'cu124' - os: macos-14 diff --git a/.github/workflows/building.yml b/.github/workflows/building.yml index 97b189f..22eb5dd 100644 --- a/.github/workflows/building.yml +++ b/.github/workflows/building.yml @@ -12,9 +12,11 @@ jobs: matrix: os: [ubuntu-20.04, macos-14, windows-2019] python-version: ['3.8', '3.9', '3.10', '3.11', '3.12'] - torch-version: [2.4.0] # [2.3.0, 2.4.0] + torch-version: [2.5.0] # [2.3.0, 2.4.0, 2.5.0] cuda-version: ['cpu', 'cu118', 'cu121', 'cu124'] exclude: + - torch-version: 2.5.0 + python-version: '3.8' - torch-version: 2.3.0 cuda-version: 'cu124' - os: macos-14 diff --git a/.github/workflows/testing.yml b/.github/workflows/testing.yml index 0406da5..070f3cb 100644 --- a/.github/workflows/testing.yml +++ b/.github/workflows/testing.yml @@ -15,8 +15,8 @@ jobs: fail-fast: false matrix: os: [ubuntu-latest, windows-latest] - python-version: [3.8] - torch-version: [2.3.0, 2.4.0] + python-version: [3.9] + torch-version: [2.4.0, 2.5.0] steps: - uses: actions/checkout@v2 @@ -30,7 +30,7 @@ jobs: pip install torch==${{ matrix.torch-version }} --extra-index-url https://download.pytorch.org/whl/cpu - name: Install scipy - if: ${{ matrix.python-version == '3.8' }} + if: ${{ matrix.python-version == '3.9' }} run: | pip install scipy==1.10.1 diff --git a/README.md b/README.md index 888e08c..1aed250 100644 --- a/README.md +++ b/README.md @@ -43,12 +43,12 @@ conda install pytorch-cluster -c pyg We alternatively provide pip wheels for all major OS/PyTorch/CUDA combinations, see [here](https://data.pyg.org/whl). -#### PyTorch 2.4 +#### PyTorch 2.5 -To install the binaries for PyTorch 2.4.0, simply run +To install the binaries for PyTorch 2.5.0, simply run ``` -pip install torch-cluster -f https://data.pyg.org/whl/torch-2.4.0+${CUDA}.html +pip install torch-cluster -f https://data.pyg.org/whl/torch-2.5.0+${CUDA}.html ``` where `${CUDA}` should be replaced by either `cpu`, `cu118`, `cu121`, or `cu124` depending on your PyTorch installation. @@ -59,24 +59,23 @@ where `${CUDA}` should be replaced by either `cpu`, `cu118`, `cu121`, or `cu124` | **Windows** | ✅ | ✅ | ✅ | ✅ | | **macOS** | ✅ | | | | +#### PyTorch 2.4 -#### PyTorch 2.3 - -To install the binaries for PyTorch 2.3.0, simply run +To install the binaries for PyTorch 2.4.0, simply run ``` -pip install torch-cluster -f https://data.pyg.org/whl/torch-2.3.0+${CUDA}.html +pip install torch-cluster -f https://data.pyg.org/whl/torch-2.4.0+${CUDA}.html ``` -where `${CUDA}` should be replaced by either `cpu`, `cu118`, or `cu121` depending on your PyTorch installation. +where `${CUDA}` should be replaced by either `cpu`, `cu118`, `cu121`, or `cu124` depending on your PyTorch installation. -| | `cpu` | `cu118` | `cu121` | -|-------------|-------|---------|---------| -| **Linux** | ✅ | ✅ | ✅ | -| **Windows** | ✅ | ✅ | ✅ | -| **macOS** | ✅ | | | +| | `cpu` | `cu118` | `cu121` | `cu124` | +|-------------|-------|---------|---------|---------| +| **Linux** | ✅ | ✅ | ✅ | ✅ | +| **Windows** | ✅ | ✅ | ✅ | ✅ | +| **macOS** | ✅ | | | | -**Note:** Binaries of older versions are also provided for PyTorch 1.4.0, PyTorch 1.5.0, PyTorch 1.6.0, PyTorch 1.7.0/1.7.1, PyTorch 1.8.0/1.8.1, PyTorch 1.9.0, PyTorch 1.10.0/1.10.1/1.10.2, PyTorch 1.11.0, PyTorch 1.12.0/1.12.1, PyTorch 1.13.0/1.13.1, PyTorch 2.0.0/2.0.1, PyTorch 2.1.0/2.1.1/2.1.2, and PyTorch 2.2.0/2.2.1/2.2.2 (following the same procedure). +**Note:** Binaries of older versions are also provided for PyTorch 1.4.0, PyTorch 1.5.0, PyTorch 1.6.0, PyTorch 1.7.0/1.7.1, PyTorch 1.8.0/1.8.1, PyTorch 1.9.0, PyTorch 1.10.0/1.10.1/1.10.2, PyTorch 1.11.0, PyTorch 1.12.0/1.12.1, PyTorch 1.13.0/1.13.1, PyTorch 2.0.0/2.0.1, PyTorch 2.1.0/2.1.1/2.1.2, PyTorch 2.2.0/2.2.1/2.2.2, and PyTorch 2.3.0/2.3.1 (following the same procedure). For older versions, you need to explicitly specify the latest supported version number or install via `pip install --no-index` in order to prevent a manual installation from source. You can look up the latest supported version number [here](https://data.pyg.org/whl). diff --git a/conda/pytorch-cluster/README.md b/conda/pytorch-cluster/README.md index 746cee5..80d3bf6 100644 --- a/conda/pytorch-cluster/README.md +++ b/conda/pytorch-cluster/README.md @@ -1,3 +1,3 @@ ``` -./build_conda.sh 3.11 2.4.0 cu118 # python, pytorch and cuda version +./build_conda.sh 3.11 2.5.0 cu118 # python, pytorch and cuda version ```