From 64e59764f96baad772814043349f21121aa5914f Mon Sep 17 00:00:00 2001 From: Chuck Tang Date: Wed, 25 Sep 2024 15:22:39 -0700 Subject: [PATCH] commit change --- .github/workflows/docker.yaml | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml index ecdbc036..60f6fb2a 100644 --- 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: