Update build.yml

ggerganov · Jan 14, 2025 · 66ec17e · 66ec17e
1 parent 41b4b11
commit 66ec17e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -850,7 +850,7 @@ jobs:
             arch: 89-real
             package: false
 
-    container: nvidia/cuda:${{ matrix.cuda.version }}-devel-ubuntu24.04
+    container: ${{ matrix.cuda.container }}
 
     name: ubuntu-22-cuda (${{ matrix.cuda.version }} cap ${{ matrix.cuda.cap }}, e.g. ${{ matrix.cuda.example }})