Remove torch 2.1 from docker build (#982)

mosaicml · Feb 16, 2024 · 3a99270 · 3a99270
1 parent da2c863
commit 3a99270
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml
@@ -17,12 +17,6 @@ jobs:
     strategy:
       matrix:
         include:
-        - name: "2.1.2_cu121_flash2"
-          base_image: mosaicml/pytorch:2.1.2_cu121-python3.10-ubuntu20.04
-          dep_groups: "[gpu-flash2]"
-        - 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]"