Support pytorch 2.2.0

.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'

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