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

Reverted Triton removal to use system triton package

Browse Source
This commit is contained in:
Eugene Rakhmatulin
2026-01-31 23:23:59 -08:00
parent c81edce091
commit 5f7d480801
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Dockerfile
View File

@@ -38,7 +38,7 @@ RUN apt update && \
curl vim ninja-build git \ curl vim ninja-build git \
ccache \ ccache \
&& rm -rf /var/lib/apt/lists/* \ && 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++ # Configure Ccache for CUDA/C++
ENV PATH=/usr/lib/ccache:$PATH ENV PATH=/usr/lib/ccache:$PATH
@@ -243,7 +243,7 @@ RUN apt update && \
curl vim git \ curl vim git \
libxcb1 \ libxcb1 \
&& rm -rf /var/lib/apt/lists/* \ && 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 # Set final working directory
WORKDIR $VLLM_BASE_DIR WORKDIR $VLLM_BASE_DIR
@@ -281,10 +281,6 @@ RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \
# Cleanup # 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 # Keeping it here for reference - this won't work as is without squashing layers
# RUN uv pip uninstall absl-py apex argon2-cffi \ # RUN uv pip uninstall absl-py apex argon2-cffi \
# argon2-cffi-bindings arrow asttokens astunparse async-lru audioread babel beautifulsoup4 \ # argon2-cffi-bindings arrow asttokens astunparse async-lru audioread babel beautifulsoup4 \