From 6aae9949a1c93e0508b4a51e541533cf7bdd11f2 Mon Sep 17 00:00:00 2001 From: Olli Lupton Date: Tue, 29 Oct 2024 08:43:06 -0700 Subject: [PATCH] CUDA: bump to 12.6.2 This includes NCCL 2.23.4. --- .github/container/Dockerfile.base | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/container/Dockerfile.base b/.github/container/Dockerfile.base index 2d0ae8509..50fda91a2 100644 --- 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 GIT_USER_EMAIL=jax@nvidia.com ARG CLANG_VERSION=18