add torch 2.3.1 base image (#1745)

axolotl-ai-cloud · Jul 13, 2024 · 137d84d · 137d84d
1 parent 18abdb4
commit 137d84d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml
@@ -37,6 +37,11 @@ jobs:
             python_version: "3.11"
             pytorch: 2.3.0
             torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
+          - cuda: "121"
+            cuda_version: 12.1.0
+            python_version: "3.11"
+            pytorch: 2.3.1
+            torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
     steps:
       - name: Checkout
         uses: actions/checkout@v3