From 41fe988c95d74ff9f22b0fef3095a3a04af853dd Mon Sep 17 00:00:00 2001 From: gmeanti Date: Mon, 12 Feb 2024 13:51:46 +0100 Subject: [PATCH] Support pytorch 2.2.0 --- .github/workflows/build_wheels.yml | 6 +++++- README.md | 14 +++++++------- 2 files changed, 12 insertions(+), 8 deletions(-) diff --git a/.github/workflows/build_wheels.yml b/.github/workflows/build_wheels.yml index ba376aa..c24497a 100644 --- a/.github/workflows/build_wheels.yml +++ b/.github/workflows/build_wheels.yml @@ -15,15 +15,19 @@ jobs: matrix: os: [ubuntu-20.04] # support version based on: https://download.pytorch.org/whl/torch/ - python-version: ['3.8', '3.9', '3.10', '3.11'] + python-version: ['3.8', '3.9', '3.10', '3.11', '3.12'] torch-version: [2.0.0, 2.1.0, 2.2.0] cuda-version: ['cpu', 'cu117', 'cu118', 'cu121'] exclude: - torch-version: 2.0.0 cuda-version: 'cu121' + - torch-version: 2.0.0 + python-version: '3.12' - torch-version: 2.1.0 cuda-version: 'cu117' + - torch-version: 2.1.0 + python-version: '3.12' - torch-version: 2.2.0 cuda-version: 'cu117' diff --git a/README.md b/README.md index 6e49c42..7d86ad6 100644 --- a/README.md +++ b/README.md @@ -42,19 +42,19 @@ pip install --no-build-isolation git+https://github.com/FalkonML/falkon.git We alternatively provide pre-built pip wheels for the following combinations of PyTorch and CUDA: -| Linux | `cu116` | `cu117` | `cu118` | `cu121` | -|--------------|---------|---------|---------|---------| -| torch 1.13.0 | ✅ | ✅ | | | -| torch 2.0.0 | | ✅ | ✅ | | -| torch 2.1.0 | | | ✅ | ✅ | +| Linux | `cu117` | `cu118` | `cu121` | +|--------------|---------|---------|---------| +| torch 2.0.0 | ✅ | ✅ | | +| torch 2.1.0 | | ✅ | ✅ | +| torch 2.2.0 | | ✅ | ✅ | For other combinations, and previous versions of Falkon, please check [here](https://falkon.dibris.unige.it/index.html) for a list of supported wheels. To install a wheel for a specific PyTorch + CUDA combination, you can run ```bash -# e.g., torch 2.1.0 + CUDA 12.1 -pip install falkon -f https://falkon.dibris.unige.it/torch-2.1.0_cu121.html +# e.g., torch 2.2.0 + CUDA 12.1 +pip install falkon -f https://falkon.dibris.unige.it/torch-2.2.0_cu121.html ``` Similarly for CPU-only packages