deprecate: pytorch 2.0.1 image (#1315) [skip ci]

* deprecate: pytorch 2.0.1 image * deprecate from main image * Update main.yml * Update tests.yml
axolotl-ai-cloud · Feb 22, 2024 · 7d23fc8 · 7d23fc8
1 parent bb1df9f
commit 7d23fc8
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 16 deletions.
diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml
@@ -12,11 +12,6 @@ jobs:
       fail-fast: false
       matrix:
         include:
-          - cuda: "118"
-            cuda_version: 11.8.0
-            python_version: "3.10"
-            pytorch: 2.0.1
-            torch_cuda_arch_list: "7.0 7.5 8.0 8.6 9.0+PTX"
           - cuda: "118"
             cuda_version: 11.8.0
             python_version: "3.10"

diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -13,11 +13,6 @@ jobs:
       fail-fast: false
       matrix:
         include:
-          - cuda: 118
-            cuda_version: 11.8.0
-            python_version: "3.10"
-            pytorch: 2.0.1
-            axolotl_extras:
           - cuda: 118
             cuda_version: 11.8.0
             python_version: "3.10"
@@ -73,11 +68,6 @@ jobs:
     strategy:
       matrix:
         include:
-          - cuda: 118
-            cuda_version: 11.8.0
-            python_version: "3.10"
-            pytorch: 2.0.1
-            axolotl_extras:
           - cuda: 118
             cuda_version: 11.8.0
             python_version: "3.10"

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -69,7 +69,7 @@ jobs:
           - cuda: 118
             cuda_version: 11.8.0
             python_version: "3.10"
-            pytorch: 2.0.1
+            pytorch: 2.1.2
           - cuda: 121
             cuda_version: 12.1.0
             python_version: "3.10"