diff --git a/Dockerfile b/Dockerfile
index 6a5ad92..c99af4a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,7 +2,7 @@ ARG CUDA_VERSION=12.8.1
 ARG PYTHON_VERSION=3.10
 
 
-ARG BUILD_BASE_IMAGE=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
+ARG BUILD_BASE_IMAGE=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04
 ARG FINAL_BASE_IMAGE=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04
 
 #################### BASE BUILD IMAGE ####################