fix

mosaicml · Dec 7, 2023 · 07ccfa9 · 07ccfa9
1 parent 0e29f8a
commit 07ccfa9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml
@@ -17,6 +17,9 @@ jobs:
     strategy:
       matrix:
         include:
+        - name: '2.1.0_cu121'	
+          base_image: mosaicml/pytorch:2.1.0_cu121-python3.10-ubuntu20.04	
+          dep_groups: '[gpu]'
         - name: '2.1.0_cu121_flash2'
           base_image: mosaicml/pytorch:2.1.0_cu121-python3.10-ubuntu20.04
           dep_groups: '[gpu-flash2]'