From aedaba8664f67d8ae1538e5e5cec2b1ce3f258dd Mon Sep 17 00:00:00 2001 From: "Kevin H. Luu" Date: Mon, 28 Sep 2026 20:17:59 -0700 Subject: [PATCH] [CI/Build] Skip the snapshot runtime on CUDA 12.x images (#59118) Signed-off-by: khluu Co-authored-by: Claude Opus 5.5 --- docker/Dockerfile | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 6609ed7464..5f4c4ab35f 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -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