A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 14, 05:30 UTC
Update base image to 22.04 (jammy) (#26065)
Signed-off-by: Huy Do <huydhn@gmail.com>
Huy Do committed
Oct 2, 2025 at 05:48 UTC
d4e7a1152d082b45f193972f34ed2a096a6b2901
3 files changed
+3
-3
docker/Dockerfile
+2
-2
@@ -13,8 +13,8 @@ ARG PYTHON_VERSION=3.12
13
# private registries that use a different repository naming conventions.
14
#
15
# Example:
16
-# docker build --build-arg BUILD_BASE_IMAGE=registry.acme.org/mirror/nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
17
-ARG BUILD_BASE_IMAGE=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
16
+# docker build --build-arg BUILD_BASE_IMAGE=registry.acme.org/mirror/nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04
17
+ARG BUILD_BASE_IMAGE=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04
18
# TODO: Restore to base image after FlashInfer AOT wheel fixed
19
ARG FINAL_BASE_IMAGE=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04
20
docker/Dockerfile.nightly_torch
+1
-1
@@ -6,7 +6,7 @@ ARG CUDA_VERSION=12.8.0
6
#
7
#################### BASE BUILD IMAGE ####################
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
ARG CUDA_VERSION=12.8.0
11
ARG PYTHON_VERSION=3.12
12
ARG TARGETPLATFORM
docs/assets/contributing/dockerfile-stages-dependency.png
Binary files a/docs/assets/contributing/dockerfile-stages-dependency.png and b/docs/assets/contributing/dockerfile-stages-dependency.png differ