Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions vllm/x86_64/gpu/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,8 @@ RUN bash install_efa.sh ${EFA_VERSION} && \
# ====================== ec2 =========================================
FROM base AS vllm-ec2

RUN apt-get update && \
RUN dpkg -l | grep -E "cuda|nvidia|libnv" | awk '{print $2}' | xargs apt-mark hold && \
apt-get update && \
apt-get upgrade -y && \
apt-get clean

Expand All @@ -73,7 +74,8 @@ ENTRYPOINT ["/usr/local/bin/dockerd_entrypoint.sh"]
# ====================== sagemaker =========================================
FROM base AS vllm-sagemaker

RUN apt-get update && \
RUN dpkg -l | grep -E "cuda|nvidia|libnv" | awk '{print $2}' | xargs apt-mark hold && \
apt-get update && \
apt-get upgrade -y && \
apt-get clean

Expand Down
4 changes: 3 additions & 1 deletion vllm/x86_64/gpu/Dockerfile.rayserve
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,9 @@ RUN bash install_efa.sh ${EFA_VERSION} && \
FROM base AS vllm-rayserve-ec2

RUN uv pip install --system ray[serve]==2.49.0 && \
uv cache clean \
uv cache clean

RUN dpkg -l | grep -E "cuda|nvidia|libnv" | awk '{print $2}' | xargs apt-mark hold && \
apt-get update && \
apt-get upgrade -y && \
apt-get clean
Expand Down