add new images

mosaicml · Feb 14, 2024 · 0afd924 · 0afd924
1 parent bce5374
commit 0afd924
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml
@@ -23,6 +23,12 @@ jobs:
         - name: "2.1.2_cu121_flash2_aws"
           base_image: mosaicml/pytorch:2.1.2_cu121-python3.10-ubuntu20.04-aws
           dep_groups: "[gpu-flash2]"
+        - name: "2.2.0_cu121_flash2"
+          base_image: mosaicml/pytorch:2.2.0_cu121-python3.11-ubuntu20.04
+          dep_groups: "[gpu-flash2]"
+        - name: "2.2.0_cu121_flash2_aws"
+          base_image: mosaicml/pytorch:2.2.0_cu121-python3.11-ubuntu20.04-aws
+          dep_groups: "[gpu-flash2]"
     steps:
     - name: Maximize Build Space on Worker
       uses: easimon/maximize-build-space@v4