support for cuda 12.1 (#989)

2023-12-22 11:08:22 -05:00
parent 7d4185ffcb
commit 37820f6540
2 changed files with 15 additions and 0 deletions
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -29,6 +29,11 @@ jobs:
            python_version: "3.10"
            pytorch: 2.1.1
            axolotl_extras:
+          - cuda: 121
+            cuda_version: 12.1.0
+            python_version: "3.10"
+            pytorch: 2.1.1
+            axolotl_extras:
    runs-on: [self-hosted, gpu, docker]
    steps:
      - name: Checkout
@@ -95,6 +100,11 @@ jobs:
            python_version: "3.10"
            pytorch: 2.1.1
            axolotl_extras:
+          - cuda: 121
+            cuda_version: 12.1.0
+            python_version: "3.10"
+            pytorch: 2.1.1
+            axolotl_extras:
    runs-on: [self-hosted, gpu, docker]
    steps:
      - name: Checkout