From 0afd924cd670ef758428d3c23ce22b279a1a4dfd Mon Sep 17 00:00:00 2001 From: Daniel King Date: Wed, 14 Feb 2024 10:17:31 -0800 Subject: [PATCH] add new images --- .github/workflows/docker.yaml | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml index 2ebbbd69f7..431fb7ddfa 100644 --- 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