From d886783494c597c084952c4ddca50e7d5c6f9413 Mon Sep 17 00:00:00 2001 From: Chuck Tang Date: Wed, 20 Nov 2024 14:56:38 -0800 Subject: [PATCH] commit change --- .github/workflows/docker.yaml | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml index 39043ef92a..b548f34234 100644 --- a/.github/workflows/docker.yaml +++ b/.github/workflows/docker.yaml @@ -17,6 +17,12 @@ jobs: strategy: matrix: include: + - 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-aws + dep_groups: "[all]" - name: "2.5.1_cu124" base_image: mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu22.04 dep_groups: "[all]"