From 41b4b11340882f7d2ff8ce75f01012b059a4c80c Mon Sep 17 00:00:00 2001
From: ochafik <ochafik@google.com>
Date: Tue, 14 Jan 2025 02:19:41 +0000
Subject: [PATCH] Fix container tags + rename

---
 .github/workflows/build.yml | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index b26c476d70eec..02056fcc4b326 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -815,44 +815,44 @@ jobs:
           # Capabilities of GPUs are listed on https://developer.nvidia.com/cuda-gpus, can test w/ `nvidia-smi --query-gpu=compute_cap --format=csv`
           # See available containers at https://hub.docker.com/r/nvidia/cuda/tags
           - version: 12.2
-            container: nvidia/cuda:12.2.2-devel-ubuntu24.04
+            container: nvidia/cuda:12.2.2-devel-ubuntu22.04
             cap: 7.5
             arch: 75-real
             example: 'T4'
             package: true
           - version: 12.2
-            container: nvidia/cuda:12.2.2-devel-ubuntu24.04
+            container: nvidia/cuda:12.2.2-devel-ubuntu22.04
             cap: 8.0
             arch: 80-real
             example: 'A100'
             package: true
           - version: 12.2
-            container: nvidia/cuda:12.2.2-devel-ubuntu24.04
+            container: nvidia/cuda:12.2.2-devel-ubuntu22.04
             cap: 8.6
             arch: 86-real
             example: 'A10'
             package: true
           - version: 12.2
-            container: nvidia/cuda:12.2.2-devel-ubuntu24.04
+            container: nvidia/cuda:12.2.2-devel-ubuntu22.04
             cap: 8.9
             arch: 89-real
             example: 'L4, L40S'
           - version: 12.2
-            container: nvidia/cuda:12.2.2-devel-ubuntu24.04
+            container: nvidia/cuda:12.2.2-devel-ubuntu22.04
             cap: 9.0
             arch: 90-real
             example: 'H100'
             package: true
           # Build only, don't package.
           - version: 12.6
-            container: nvidia/cuda:12.6.2-devel-ubuntu24.04
+            container: nvidia/cuda:12.6.2-devel-ubuntu22.04
             cap: 8.9
             arch: 89-real
             package: false
 
     container: nvidia/cuda:${{ matrix.cuda.version }}-devel-ubuntu24.04
 
-    name: Linux CUDA ${{ matrix.cuda.version }} Cap ${{ matrix.cuda.cap }} (e.g. ${{ matrix.cuda_cap.example_gpu }})
+    name: ubuntu-22-cuda (${{ matrix.cuda.version }} cap ${{ matrix.cuda.cap }}, e.g. ${{ matrix.cuda.example }})
 
     steps:
         - name: Clone