diff --git a/Dockerfile b/Dockerfile
index 820ae94a..ed0be490 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,7 @@
-FROM nvidia/cuda:12.4.1-base-ubuntu22.04
+#Original
+#FROM nvidia/cuda:12.4.1-base-ubuntu22.04
+#Cuda 12.8/13
+FROM nvidia/cuda:12.8.1-base-ubuntu22.04
 ENV DEBIAN_FRONTEND noninteractive
 ENV CMDARGS --listen