Add ci and images for CUDA 12.8 for B200s (#2683) [skip ci]

* Add ci and images for CUDA 12.8 for B200s * add comments explaining CI [skip e2e]
2025-05-16 13:06:08 -04:00
parent 86472715da
commit 8f8a7afb05
2 changed files with 19 additions and 0 deletions
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -31,6 +31,11 @@ jobs:
            python_version: "3.11"
            pytorch: 2.7.0
            axolotl_extras:
+          - cuda: 128
+            cuda_version: 12.8.1
+            python_version: "3.11"
+            pytorch: 2.7.0
+            axolotl_extras:
    runs-on: axolotl-gpu-runner
    steps:
      - name: Checkout
@@ -94,6 +99,11 @@ jobs:
            python_version: "3.11"
            pytorch: 2.7.0
            axolotl_extras:
+          - cuda: 128
+            cuda_version: 12.8.1
+            python_version: "3.11"
+            pytorch: 2.7.0
+            axolotl_extras:
    runs-on: axolotl-gpu-runner
    steps:
      - name: Checkout