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}