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 <[email protected]>
containers · Jul 2, 2024 · 0548faf · 0548faf
1 parent aee1c80
commit 0548faf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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}