[CI/Build] Skip the snapshot runtime on CUDA 12.x images (#59118)
Signed-off-by: khluu <khluu000@gmail.com> Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com>
Kevin H. Luu committed
Sep 28, 2026 at 20:17 UTC
aedaba8664f67d8ae1538e5e5cec2b1ce3f258dd
1 file changed
+6
-3
docker/Dockerfile
+6
-3
@@ -880,12 +880,15 @@ RUN if [ "${INSTALL_RUBIN_PRERELEASE}" = "true" ]; then \
880
# `vllm snapshot`. CRIU and its CUDA plugin are built from the same pinned
881
# source. The `cuda-checkpoint`-compatible helper is built from NVIDIA Dynamo's
882
# Apache-2.0 source against the public CUDA Driver checkpoint APIs instead of
883
-# redistributing NVIDIA's proprietary cuda-checkpoint binary.
883
+# redistributing NVIDIA's proprietary cuda-checkpoint binary. The helper uses
884
+# checkpoint APIs (CUcheckpointGpuPair) that first appear in the CUDA 13
885
+# headers, so CUDA 12.x images skip the snapshot runtime.
886
COPY tools/install_snapshot_runtime.sh /usr/local/bin/install_snapshot_runtime.sh
885
-RUN if [ "${TARGETPLATFORM}" = "linux/amd64" ]; then \
887
+RUN if [ "${TARGETPLATFORM}" = "linux/amd64" ] \
888
+ && [ "${CUDA_VERSION%%.*}" -ge 13 ]; then \
889
MAX_JOBS="${max_jobs}" /usr/local/bin/install_snapshot_runtime.sh; \
890
else \
888
- echo "Initialized snapshots are not installed for ${TARGETPLATFORM}"; \
891
+ echo "Initialized snapshots are not installed for ${TARGETPLATFORM} with CUDA ${CUDA_VERSION}"; \
892
fi
893
894
ENV CRIU_CUDA_PLUGIN_DIR=/usr/local/lib/criu