update

pyg-team · Nov 11, 2024 · 5c85d03 · 5c85d03
1 parent fd377c9
commit 5c85d03
Show file tree

Hide file tree

Showing 5 changed files with 27 additions and 5 deletions.
diff --git a/.github/actions/setup/action.yml b/.github/actions/setup/action.yml
@@ -6,7 +6,7 @@ inputs:
     default: '3.9'
   torch-version:
     required: false
-    default: '2.4.0'
+    default: '2.5.0'
   cuda-version:
     required: false
     default: cpu

diff --git a/.github/workflows/aws/upload_nightly_index.py b/.github/workflows/aws/upload_nightly_index.py
@@ -35,6 +35,10 @@
         wheels_dict[torch_version.replace('2.2.0', '2.2.2')].append(wheel)
     if '2.3.0' in torch_version:
         wheels_dict[torch_version.replace('2.3.0', '2.3.1')].append(wheel)
+    if '2.4.0' in torch_version:
+        wheels_dict[torch_version.replace('2.4.0', '2.4.1')].append(wheel)
+    if '2.5.0' in torch_version:
+        wheels_dict[torch_version.replace('2.5.0', '2.5.1')].append(wheel)
 
 index_html = html.format('\n'.join([
     href.format(f'{version}.html'.replace('+', '%2B'), version)

diff --git a/.github/workflows/building.yml b/.github/workflows/building.yml
@@ -12,8 +12,8 @@ jobs:
       matrix:
         os: [ubuntu-20.04, macos-14, windows-2019]
         python-version: ['3.9', '3.10', '3.11', '3.12']
-        # torch-version: [1.12.0, 1.13.0, 2.0.0, 2.1.0, 2.2.0, 2.3.0, 2.4.0]
-        torch-version: [2.4.0]
+        # torch-version: [1.12.0, 1.13.0, 2.0.0, 2.1.0, 2.2.0, 2.3.0, 2.4.0, 2.5.0]
+        torch-version: [2.5.0]
         cuda-version: ['cpu', 'cu113', 'cu116', 'cu117', 'cu118', 'cu121', 'cu124']
         exclude:
           - torch-version: 1.12.0
@@ -80,6 +80,12 @@ jobs:
             cuda-version: 'cu116'
           - torch-version: 2.4.0
             cuda-version: 'cu117'
+          - torch-version: 2.5.0
+            cuda-version: 'cu113'
+          - torch-version: 2.5.0
+            cuda-version: 'cu116'
+          - torch-version: 2.5.0
+            cuda-version: 'cu117'
           - os: macos-14
             cuda-version: 'cu113'
           - os: macos-14

diff --git a/.github/workflows/nightly.yml b/.github/workflows/nightly.yml
@@ -16,7 +16,7 @@ jobs:
       matrix:
         os: [ubuntu-20.04, macos-14, windows-2019]
         python-version: ['3.9', '3.10', '3.11', '3.12']
-        torch-version: [1.12.0, 1.13.0, 2.0.0, 2.1.0, 2.2.0, 2.3.0, 2.4.0]
+        torch-version: [1.12.0, 1.13.0, 2.0.0, 2.1.0, 2.2.0, 2.3.0, 2.4.0, 2.5.0]
         cuda-version: ['cpu', 'cu113', 'cu116', 'cu117', 'cu118', 'cu121', 'cu124']
         exclude:
           - torch-version: 1.12.0
@@ -83,6 +83,12 @@ jobs:
             cuda-version: 'cu116'
           - torch-version: 2.4.0
             cuda-version: 'cu117'
+          - torch-version: 2.5.0
+            cuda-version: 'cu113'
+          - torch-version: 2.5.0
+            cuda-version: 'cu116'
+          - torch-version: 2.5.0
+            cuda-version: 'cu117'
           - os: macos-14
             cuda-version: 'cu113'
           - os: macos-14

diff --git a/README.md b/README.md
@@ -29,11 +29,17 @@ pip install pyg-lib -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
 
 where
 
-* `${TORCH}` should be replaced by either `1.12.0`, `1.13.0`, `2.0.0`, `2.1.0`, `2.2.0`, `2.3.0`, or `2.4.0`
+* `${TORCH}` should be replaced by either `1.12.0`, `1.13.0`, `2.0.0`, `2.1.0`, `2.2.0`, `2.3.0`, `2.4.0` or `2.5.0`
 * `${CUDA}` should be replaced by either `cpu`, `cu102`, `cu113`, `cu116`, `cu117`, `cu118`, or `cu121`
 
 The following combinations are supported:
 
+| PyTorch 2.5  | `cpu` | `cu113` | `cu116` | `cu117` | `cu118` | `cu121` | `cu124` |
+|--------------|-------|---------|---------|---------|---------|---------|---------|
+| **Linux**    | ✅    |         |         |         | ✅      | ✅      | ✅      |
+| **Windows**  | ✅    |         |         |         | ✅      | ✅      | ✅      |
+| **macOS**    | ✅    |         |         |         |         |         |         |
+
 | PyTorch 2.4  | `cpu` | `cu113` | `cu116` | `cu117` | `cu118` | `cu121` | `cu124` |
 |--------------|-------|---------|---------|---------|---------|---------|---------|
 | **Linux**    | ✅    |         |         |         | ✅      | ✅      | ✅      |