diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index f73836f..3e408de 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.0.1-base-ubuntu22.04
+FROM nvidia/cuda:12.3.1-base-ubuntu22.04
 
 ENV PYTHON_VERSION=3.10