From 0548faf9b5de18b8eebb4b6f62f947b0d283cf31 Mon Sep 17 00:00:00 2001 From: Fabien Dupont Date: Tue, 2 Jul 2024 08:52:44 -0400 Subject: [PATCH] Update NVIDIA driver and CUDA in nvidia-bootc For the InstructLab image, we use NVIDIA driver version `550.90.07` with CUDA `12.4.1`, so this change updates the versions in the bootc image to align the stack. Signed-off-by: Fabien Dupont --- training/nvidia-bootc/Containerfile | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/training/nvidia-bootc/Containerfile b/training/nvidia-bootc/Containerfile index 16c1962f..fa8473b7 100644 --- a/training/nvidia-bootc/Containerfile +++ b/training/nvidia-bootc/Containerfile @@ -15,7 +15,7 @@ ARG KERNEL_VERSION='' ARG BUILD_ARCH='' ARG TARGET_ARCH='' -ARG DRIVER_VERSION='550.54.15' +ARG DRIVER_VERSION='550.90.07' USER builder @@ -79,9 +79,9 @@ ARG KERNEL_VERSION='' ARG DRIVER_TYPE=passthrough ENV NVIDIA_DRIVER_TYPE=${DRIVER_TYPE} -ARG DRIVER_VERSION='550.54.15' +ARG DRIVER_VERSION='550.90.07' ENV NVIDIA_DRIVER_VERSION=${DRIVER_VERSION} -ARG CUDA_VERSION='12.3.2' +ARG CUDA_VERSION='12.4.1' ARG TARGET_ARCH='' ENV TARGETARCH=${TARGET_ARCH}