From 5f7d4808011d3c92a187bd1e0ff31711bead1fe7 Mon Sep 17 00:00:00 2001 From: Eugene Rakhmatulin Date: Sat, 31 Jan 2026 23:23:59 -0800 Subject: [PATCH] Reverted Triton removal to use system triton package --- Dockerfile | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/Dockerfile b/Dockerfile index 48572b1..218b20b 100644 --- a/Dockerfile +++ b/Dockerfile @@ -38,7 +38,7 @@ RUN apt update && \ curl vim ninja-build git \ ccache \ && rm -rf /var/lib/apt/lists/* \ - && pip install uv && pip uninstall -y flash-attn pytorch-triton triton-kernels + && pip install uv && pip uninstall -y flash-attn # Configure Ccache for CUDA/C++ ENV PATH=/usr/lib/ccache:$PATH @@ -243,7 +243,7 @@ RUN apt update && \ curl vim git \ libxcb1 \ && rm -rf /var/lib/apt/lists/* \ - && pip install uv && pip uninstall -y flash-attn pytorch-triton triton-kernels + && pip install uv && pip uninstall -y flash-attn triton-kernels # pytorch-triton # Set final working directory WORKDIR $VLLM_BASE_DIR @@ -281,10 +281,6 @@ RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \ # Cleanup -# If not compiling Triton -# remove triton-kernels as they are not compatible with this vLLM version yet -RUN uv pip uninstall triton-kernels - # Keeping it here for reference - this won't work as is without squashing layers # RUN uv pip uninstall absl-py apex argon2-cffi \ # argon2-cffi-bindings arrow asttokens astunparse async-lru audioread babel beautifulsoup4 \