software-engineering/spark-vllm-docker
2
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Bumped base image version

Browse Source
This commit is contained in:
Eugene Rakhmatulin
2026-03-03 23:27:47 -08:00
parent 7d8465fd9c
commit ae19b66fdd
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile
View File

@@ -6,7 +6,7 @@ ARG BUILD_JOBS=16
# ========================================================= # =========================================================
# STAGE 1: Base Image (Installs Dependencies) # STAGE 1: Base Image (Installs Dependencies)
# ========================================================= # =========================================================
FROM nvcr.io/nvidia/pytorch:26.01-py3 AS base FROM nvcr.io/nvidia/pytorch:26.02-py3 AS base
# Build parallemism # Build parallemism
ARG BUILD_JOBS ARG BUILD_JOBS
@@ -207,7 +207,7 @@ COPY --from=vllm-builder /workspace/wheels /
# ========================================================= # =========================================================
# STAGE 6: Runner (Installs wheels from host ./wheels/) # STAGE 6: Runner (Installs wheels from host ./wheels/)
# ========================================================= # =========================================================
FROM nvcr.io/nvidia/pytorch:26.01-py3 AS runner FROM nvcr.io/nvidia/pytorch:26.02-py3 AS runner
# Transferring build settings from build image because of ptxas/jit compilation during vLLM startup # Transferring build settings from build image because of ptxas/jit compilation during vLLM startup
# Build parallemism # Build parallemism