CUDA: bump to 12.6.2 (#1127)

- This includes NCCL 2.23.4.
NVIDIA · Nov 6, 2024 · 61d8446 · 61d8446
1 parent 3e1fb6d
commit 61d8446
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/container/Dockerfile.base b/.github/container/Dockerfile.base
@@ -1,5 +1,5 @@
 # syntax=docker/dockerfile:1-labs
-ARG BASE_IMAGE=nvidia/cuda:12.6.1-devel-ubuntu22.04
+ARG BASE_IMAGE=nvidia/cuda:12.6.2-devel-ubuntu22.04
 ARG GIT_USER_NAME="JAX Toolbox"
 ARG [email protected]
 ARG CLANG_VERSION=18