diff --git a/docker/cuda.Dockerfile b/docker/cuda.Dockerfile
index a9d44ae..7fa877d 100644
--- a/docker/cuda.Dockerfile
+++ b/docker/cuda.Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.2.2-devel-ubuntu20.04
+FROM nvidia/cuda:12.4.1-devel-ubuntu20.04
 
 ENV ARGOS_DEVICE_TYPE cuda
 ARG with_models=false