Add PyTorch 2.5 support (#389)

update
rusty1s · Nov 12, 2024 · cdbf561 · cdbf561
1 parent bd56751
commit cdbf561
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 20 deletions.
diff --git 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
@@ -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
@@ -15,8 +15,8 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, windows-2019]
-        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
@@ -47,7 +47,7 @@ jobs:
           bash .github/workflows/metis-${{ runner.os }}.sh
 
       - 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
@@ -43,12 +43,12 @@ conda install pytorch-sparse -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-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.4.0+${CUDA}.html
+pip install torch-scatter torch-sparse -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-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.3.0+${CUDA}.html
+pip install torch-scatter torch-sparse -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-sparse/README.md b/conda/pytorch-sparse/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
 ```