diff --git a/Dockerfile.base b/Dockerfile.base
index a9d0900..5952ef1 100644
--- a/Dockerfile.base
+++ b/Dockerfile.base
@@ -1,5 +1,4 @@
-# syntax=docker/dockerfile:1
-FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu20.04
+FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
 ARG DEBIAN_FRONTEND="noninteractive"
 ENV DEBIAN_FRONTEND=${DEBIAN_FRONTEND}
 ENV MAMBA_ROOT_PREFIX=~/micromamba
@@ -75,4 +74,4 @@ channels:
   - defaults
 show_channel_urls: true
 EOF
-EOT
\ No newline at end of file
+EOT