From 3a9927087bed496da58b3f273a5033fa7f400db8 Mon Sep 17 00:00:00 2001 From: Daniel King <43149077+dakinggg@users.noreply.github.com> Date: Thu, 15 Feb 2024 18:38:32 -0800 Subject: [PATCH] Remove torch 2.1 from docker build (#982) --- .github/workflows/docker.yaml | 6 ------ 1 file changed, 6 deletions(-) diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml index 431fb7ddfa..c4c1937d62 100644 --- 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]"