From c81edce0911e0246ac06cf2396cf51f761a73dd9 Mon Sep 17 00:00:00 2001 From: Eugene Rakhmatulin Date: Sat, 31 Jan 2026 16:12:33 -0800 Subject: [PATCH] bumped up MXFP4 base image version --- Dockerfile.mxfp4 | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/Dockerfile.mxfp4 b/Dockerfile.mxfp4 index b1ed0ff..c218952 100644 --- a/Dockerfile.mxfp4 +++ b/Dockerfile.mxfp4 @@ -6,7 +6,7 @@ ARG BUILD_JOBS=16 # ========================================================= # STAGE 1: Base Image (Installs Dependencies) # ========================================================= -FROM nvcr.io/nvidia/pytorch:25.12-py3 AS base +FROM nvcr.io/nvidia/pytorch:26.01-py3 AS base # Build parallemism ARG BUILD_JOBS @@ -225,7 +225,7 @@ RUN --mount=type=cache,id=ccache,target=/root/.ccache \ # ========================================================= # STAGE 4: Runner (Transfers only necessary artifacts) # ========================================================= -FROM nvcr.io/nvidia/pytorch:25.12-py3 AS runner +FROM nvcr.io/nvidia/pytorch:26.01-py3 AS runner ENV DEBIAN_FRONTEND=noninteractive ENV PIP_BREAK_SYSTEM_PACKAGES=1 @@ -275,3 +275,6 @@ RUN chmod +x $VLLM_BASE_DIR/run-cluster-node.sh RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \ uv pip install ray[default] fastsafetensors +# If not compiling Triton +# remove triton-kernels as they are not compatible with this vLLM version yet +RUN uv pip uninstall triton-kernels