iHore/vLLM
8
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[CI/Build] Skip the snapshot runtime on CUDA 12.x images (#59118)

Browse Source 
Signed-off-by: khluu <khluu000@gmail.com>
Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
Kevin H. Luu
2026-09-28 20:17:59 -07:00
committed by GitHub
co-authored by Claude Opus 5.5
parent 4e59d53ab5
commit aedaba8664
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+6 -3
View File
@@ -880,12 +880,15 @@ RUN if [ "${INSTALL_RUBIN_PRERELEASE}" = "true" ]; then \
# `vllm snapshot`. CRIU and its CUDA plugin are built from the same pinned
# source. The `cuda-checkpoint`-compatible helper is built from NVIDIA Dynamo's
# Apache-2.0 source against the public CUDA Driver checkpoint APIs instead of
# redistributing NVIDIA's proprietary cuda-checkpoint binary.
# redistributing NVIDIA's proprietary cuda-checkpoint binary. The helper uses
# checkpoint APIs (CUcheckpointGpuPair) that first appear in the CUDA 13
# headers, so CUDA 12.x images skip the snapshot runtime.
COPY tools/install_snapshot_runtime.sh /usr/local/bin/install_snapshot_runtime.sh
RUN if [ "${TARGETPLATFORM}" = "linux/amd64" ]; then \
RUN if [ "${TARGETPLATFORM}" = "linux/amd64" ] \
&& [ "${CUDA_VERSION%%.*}" -ge 13 ]; then \
MAX_JOBS="${max_jobs}" /usr/local/bin/install_snapshot_runtime.sh; \
else \
echo "Initialized snapshots are not installed for ${TARGETPLATFORM}"; \
echo "Initialized snapshots are not installed for ${TARGETPLATFORM} with CUDA ${CUDA_VERSION}"; \
fi
ENV CRIU_CUDA_PLUGIN_DIR=/usr/local/lib/criu