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

Updated Dockerfile with additional deps

Browse Source
This commit is contained in:
eugr
2025-11-24 15:47:20 -08:00
parent f5141974ae
commit d3fd2e69fd
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile
View File

@@ -13,6 +13,7 @@ ENV VLLM_BASE_DIR=/workspace/vllm
# Added 'git', 'wget', and 'python3-pip' as they are required for the script steps # Added 'git', 'wget', and 'python3-pip' as they are required for the script steps
RUN apt-get update && apt-get install -y \ RUN apt-get update && apt-get install -y \
curl \ curl \
vim \
cmake \ cmake \
build-essential \ build-essential \
ninja-build \ ninja-build \
@@ -20,6 +21,7 @@ RUN apt-get update && apt-get install -y \
python3-pip \ python3-pip \
git \ git \
wget \ wget \
gnuplot \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
# Setup Workspace # Setup Workspace
@@ -47,7 +49,7 @@ ARG CACHEBUST_DEPS=1
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130 RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130
# Install Helper libraries # Install Helper libraries
RUN pip install xgrammar triton RUN pip install xgrammar triton termplotlib
# Install FlashInfer # Install FlashInfer
# Note: Using the same index URLs as provided in your script # Note: Using the same index URLs as provided in your script