From 6fe699b8b9361d2612de170ab28b0e5fc67c7cc4 Mon Sep 17 00:00:00 2001 From: "Yu-Hang \"Maxin\" Tang" Date: Thu, 30 Jan 2025 07:47:15 -0800 Subject: [PATCH] Bump base image to 25.01 (CUDA 12.8) (#1276) --- .github/container/Dockerfile.base | 2 +- .github/workflows/ci.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/container/Dockerfile.base b/.github/container/Dockerfile.base index 108874a70..846c2f4b1 100644 --- a/.github/container/Dockerfile.base +++ b/.github/container/Dockerfile.base @@ -1,5 +1,5 @@ # syntax=docker/dockerfile:1-labs -ARG BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:24.11-cuda12.6-devel-ubuntu24.04 +ARG BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:25.01-cuda12.8-devel-ubuntu24.04 ARG GIT_USER_NAME="JAX Toolbox" ARG GIT_USER_EMAIL=jax@nvidia.com ARG CLANG_VERSION=18 diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index 0c3c8bdb0..52934114a 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -30,7 +30,7 @@ on: required: false CUDA_IMAGE: type: string - description: CUDA image to use as base, e.g. nvidia/cuda:X.Y.Z-devel-ubuntu22.04 + description: CUDA image to use as base, e.g. nvcr.io/nvidia/cuda-dl-base:25.01-cuda12.8-devel-ubuntu24.04 default: 'latest' required: false SOURCE_OVERRIDES: