commit change

mosaicml · Sep 25, 2024 · 64e5976 · 64e5976
1 parent 3880bb6
commit 64e5976
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml
@@ -17,11 +17,17 @@ jobs:
     strategy:
       matrix:
         include:
+        - name: "2.1.2_cu121"
+          base_image: mosaicml/pytorch:2.1.2_cu121-python3.10-ubuntu20.04
+          dep_groups: "[all]"
+        - name: "2.1.2_cu121_aws"
+          base_image: mosaicml/pytorch:2.1.2_cu121-python3.10-ubuntu20.04-aws
+          dep_groups: "[all]"
         - name: "2.4.0_cu124"
           base_image: mosaicml/pytorch:2.4.0_cu124-python3.11-ubuntu20.04
           dep_groups: "[all]"
         - name: "2.4.0_cu124_aws"
-          base_image: mosaicml/pytorch:2.4.0_cu124-python3.11-ubuntu20.04
+          base_image: mosaicml/pytorch:2.4.0_cu124-python3.11-ubuntu20.04-aws
           dep_groups: "[all]"
     steps: