Update base image to 22.04 (jammy) (#26065)

huydhn · simon-mo · commit 05bf0c52a12c · 2025-10-02T10:35:51.000-07:00
Signed-off-by: Huy Do &lt;huydhn@gmail.com&gt;
Signed-off-by: simon-mo &lt;simon.mo@hey.com&gt;
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -13,8 +13,8 @@ ARG PYTHON_VERSION=3.12
 # private registries that use a different repository naming conventions.
 #
 # Example:
-# docker build --build-arg BUILD_BASE_IMAGE=registry.acme.org/mirror/nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
-ARG BUILD_BASE_IMAGE=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
+# docker build --build-arg BUILD_BASE_IMAGE=registry.acme.org/mirror/nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04
+ARG BUILD_BASE_IMAGE=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04
 # TODO: Restore to base image after FlashInfer AOT wheel fixed
 ARG FINAL_BASE_IMAGE=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04
 
diff --git a/docker/Dockerfile.nightly_torch b/docker/Dockerfile.nightly_torch
@@ -6,7 +6,7 @@ ARG CUDA_VERSION=12.8.0
 #
 #################### BASE BUILD IMAGE ####################
 # prepare basic build environment
-FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04 AS base
+FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04 AS base
 ARG CUDA_VERSION=12.8.0
 ARG PYTHON_VERSION=3.12
 ARG TARGETPLATFORM
diff --git a/docs/assets/contributing/dockerfile-stages-dependency.png b/docs/assets/contributing/dockerfile-stages-dependency.png

Original file line number	Diff line number	Diff line change
`@@ -6,7 +6,7 @@ ARG CUDA_VERSION=12.8.0`
`6`	`6`	`#`
`7`	`7`	`#################### BASE BUILD IMAGE ####################`
`8`	`8`	`# prepare basic build environment`
`9`		`-FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04 AS base`
	`9`	`+FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04 AS base`
`10`	`10`	`ARG CUDA_VERSION=12.8.0`
`11`	`11`	`ARG PYTHON_VERSION=3.12`
`12`	`12`	`ARG TARGETPLATFORM`