Update pytorch versions

FalkonML · Feb 12, 2024 · 5bed1e9 · 5bed1e9
1 parent 69ec6e4
commit 5bed1e9
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 25 deletions.
diff --git a/.github/workflows/build_docs.yml b/.github/workflows/build_docs.yml
@@ -11,7 +11,7 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python-version: ['3.8']
-        torch-version: ['2.1.0']
+        torch-version: ['2.2.0']
 
     steps:
       - uses: actions/checkout@v3

diff --git a/.github/workflows/build_wheels.yml b/.github/workflows/build_wheels.yml
@@ -16,34 +16,16 @@ jobs:
         os: [ubuntu-20.04]
         # support version based on: https://download.pytorch.org/whl/torch/
         python-version: ['3.8', '3.9', '3.10', '3.11']
-        torch-version: [1.13.0, 2.0.0, 2.1.0]
-        cuda-version: ['cpu', 'cu113', 'cu115', 'cu116', 'cu117', 'cu118', 'cu121']
+        torch-version: [2.0.0, 2.1.0, 2.2.0]
+        cuda-version: ['cpu', 'cu117', 'cu118', 'cu121']
         exclude:
-          - torch-version: 1.13.0
-            cuda-version: 'cu113'
-          - torch-version: 1.13.0
-            cuda-version: 'cu115'
-          - torch-version: 1.13.0
-            cuda-version: 'cu118'
-          - torch-version: 1.13.0
-            cuda-version: 'cu121'
-
-          - torch-version: 2.0.0
-            cuda-version: 'cu113'
-          - torch-version: 2.0.0
-            cuda-version: 'cu115'
-          - torch-version: 2.0.0
-            cuda-version: 'cu116'
           - torch-version: 2.0.0
             cuda-version: 'cu121'
 
           - torch-version: 2.1.0
-            cuda-version: 'cu113'
-          - torch-version: 2.1.0
-            cuda-version: 'cu115'
-          - torch-version: 2.1.0
-            cuda-version: 'cu116'
-          - torch-version: 2.1.0
+            cuda-version: 'cu117'
+
+          - torch-version: 2.2.0
             cuda-version: 'cu117'
 
     steps:

diff --git a/.github/workflows/testing.yml b/.github/workflows/testing.yml
@@ -10,7 +10,7 @@ jobs:
       matrix:
         os: [ubuntu-20.04]
         python-version: ['3.8']
-        torch-version: [2.0.0, 2.1.0]
+        torch-version: [2.0.0, 2.2.0]
 
     steps:
       - name: Checkout with submodules