From 5c85d03f481eeeda2143cd665a69eb7dcc79667d Mon Sep 17 00:00:00 2001 From: rusty1s Date: Mon, 11 Nov 2024 18:53:18 +0000 Subject: [PATCH] update --- .github/actions/setup/action.yml | 2 +- .github/workflows/aws/upload_nightly_index.py | 4 ++++ .github/workflows/building.yml | 10 ++++++++-- .github/workflows/nightly.yml | 8 +++++++- README.md | 8 +++++++- 5 files changed, 27 insertions(+), 5 deletions(-) diff --git a/.github/actions/setup/action.yml b/.github/actions/setup/action.yml index 99e06303..1e91451d 100644 --- 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 index ddbb38ff..50264376 100644 --- 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 index 34a47c6c..df72269a 100644 --- 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 index b0dc0f23..7155e61c 100644 --- 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 index 00832eee..80293a40 100644 --- 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** | ✅ | | | | ✅ | ✅ | ✅ |