diff --git a/templates/docker/nvidia.dockerfile b/templates/docker/nvidia.dockerfile
index 9ff09e9f..c7b66597 100644
--- a/templates/docker/nvidia.dockerfile
+++ b/templates/docker/nvidia.dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.7.1-base-UBUNTU_DUMMY_VERSION
+FROM nvidia/cuda:12.4.1-base-UBUNTU_DUMMY_VERSION
 
 #Arguments maybe add defaults?
 ARG user