From d82cd131fa5b7ef8b87a6bb915e87de8a94f804d Mon Sep 17 00:00:00 2001 From: Gabriele Oliaro Date: Wed, 16 Oct 2024 10:58:23 -0400 Subject: [PATCH] update to 12.1 (#1512) --- .github/workflows/gpu-ci.yml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/gpu-ci.yml b/.github/workflows/gpu-ci.yml index 6ca50027d1..9ee4693f91 100644 --- a/.github/workflows/gpu-ci.yml +++ b/.github/workflows/gpu-ci.yml @@ -56,7 +56,7 @@ jobs: CONDA: "3" needs: gpu-ci-concierge container: - image: ghcr.io/flexflow/flexflow-environment-cuda-11.8:latest + image: ghcr.io/flexflow/flexflow-environment-cuda-12.1:latest options: --gpus all --shm-size=8192m steps: - name: Keep alive @@ -75,7 +75,7 @@ jobs: CONDA: "3" needs: gpu-ci-concierge container: - image: ghcr.io/flexflow/flexflow-environment-cuda-11.8:latest + image: ghcr.io/flexflow/flexflow-environment-cuda-12.1:latest options: --gpus all --shm-size=8192m steps: - name: Install updated git version @@ -151,7 +151,7 @@ jobs: HUGGINGFACE_TOKEN: ${{ secrets.HUGGINGFACE_TOKEN }} needs: gpu-ci-concierge container: - image: ghcr.io/flexflow/flexflow-environment-cuda-11.8:latest + image: ghcr.io/flexflow/flexflow-environment-cuda-12.1:latest options: --gpus all --shm-size=8192m steps: - name: Install updated git version @@ -239,7 +239,7 @@ jobs: CONDA: "3" needs: inference-tests container: - image: ghcr.io/flexflow/flexflow-environment-cuda-11.8:latest + image: ghcr.io/flexflow/flexflow-environment-cuda-12.1:latest options: --gpus all --shm-size=8192m steps: - name: Install updated git version