Merge pull request #653 from fabiendupont/nvidia-align-driver-and-cuda

Update NVIDIA driver and CUDA in nvidia-bootc
containers · Jul 4, 2024 · 67e8135 · 67e8135
2 parents 610d8ec + 0548faf
commit 67e8135
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}