diff --git a/Dockerfile.cuda b/Dockerfile.cuda
index 8d6f3c7..9f6ce17 100644
--- a/Dockerfile.cuda
+++ b/Dockerfile.cuda
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
+FROM nvidia/cuda:12.5.0-runtime-ubuntu22.04
 WORKDIR /app
 COPY . /app
 RUN apt-get update && \