diff --git a/docker/Dockerfile.train b/docker/Dockerfile.train
index d22212c..b020bc1 100644
--- a/docker/Dockerfile.train
+++ b/docker/Dockerfile.train
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04 as cuda-base
+FROM nvidia/cuda:12.3.0-runtime-ubuntu22.04 as cuda-base
 
 SHELL ["/bin/bash", "-o", "pipefail", "-c"]