diff --git a/deployments/container/Dockerfile.ubi8 b/deployments/container/Dockerfile.ubi8
index 9c2d7706..2688e254 100644
--- a/deployments/container/Dockerfile.ubi8
+++ b/deployments/container/Dockerfile.ubi8
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-FROM nvcr.io/nvidia/cuda:12.5.0-base-ubi8 as build
+FROM nvcr.io/nvidia/cuda:12.5.1-base-ubi8 as build
 
 SHELL ["/bin/bash", "-c"]
 
@@ -58,7 +58,7 @@ RUN go install -ldflags "-extldflags=-Wl,-z,lazy -s -w" \
     github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk@${NVIDIA_CTK_VERSION} \
     && cp ${GOPATH}/bin/nvidia-ctk /artifacts/nvidia-ctk
 
-FROM nvcr.io/nvidia/cuda:12.5.0-base-ubi8
+FROM nvcr.io/nvidia/cuda:12.5.1-base-ubi8
 
 # Remove CUDA libs(compat etc) in favor of libs installed by the NVIDIA driver
 RUN dnf remove -y cuda-*
diff --git a/deployments/container/Dockerfile.ubuntu b/deployments/container/Dockerfile.ubuntu
index 65630809..6505d861 100644
--- a/deployments/container/Dockerfile.ubuntu
+++ b/deployments/container/Dockerfile.ubuntu
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-FROM nvcr.io/nvidia/cuda:12.5.0-base-ubuntu20.04 as build
+FROM nvcr.io/nvidia/cuda:12.5.1-base-ubuntu20.04 as build
 
 SHELL ["/bin/bash", "-c"]
 
@@ -58,7 +58,7 @@ RUN go install -ldflags "-extldflags=-Wl,-z,lazy -s -w" \
     github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk@${NVIDIA_CTK_VERSION} \
     && cp ${GOPATH}/bin/nvidia-ctk /artifacts/nvidia-ctk
 
-FROM nvcr.io/nvidia/cuda:12.5.0-base-ubuntu20.04
+FROM nvcr.io/nvidia/cuda:12.5.1-base-ubuntu20.04
 
 # Remove CUDA libs(compat etc) in favor of libs installed by the NVIDIA driver
 RUN apt-get --purge -y autoremove cuda-*