diff --git a/docker/cuda.Dockerfile b/docker/cuda.Dockerfile
index 9ce0751..e08c87f 100644
--- a/docker/cuda.Dockerfile
+++ b/docker/cuda.Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.4.1-devel-ubuntu20.04
+FROM nvidia/cuda:12.4.1-devel-ubuntu22.04
 
 ENV ARGOS_DEVICE_TYPE auto
 ARG with_models=false