@tduncan / Nvidia-Optimization-Stack-VLLM Mirror · read-only / commits / af89140efc
Fork 0

A high-throughput and memory-efficient inference and serving engine for LLMs

Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git

Synced Nov 5, 21:00 UTC

[ROCm][CI] Fix UV install in Dockerfile.rocm to detect curl failures and retry (#38415)

Signed-off-by: Andreas Karatzas <akaratza@amd.com>

Andreas Karatzas committed Mar 28, 2026 at 11:47 UTC af89140efc06c462ae531999b9f2db6ba0c7a528
1 file changed +5 -2
docker/Dockerfile.rocm
+5 -2
@@ -29,8 +29,11 @@ RUN if [ "$USE_SCCACHE" != "1" ]; then \
29 rm -f "$(which sccache)" || true; \
30 fi
31
32 -# Install UV
33 -RUN curl -LsSf https://astral.sh/uv/install.sh | env UV_INSTALL_DIR="/usr/local/bin" sh
32 +# Install UV — download first, then run, so a curl failure is not masked by the pipe
33 +RUN curl -LsSf --retry 3 --retry-delay 5 https://astral.sh/uv/install.sh -o /tmp/uv-install.sh \
34 + && env UV_INSTALL_DIR="/usr/local/bin" sh /tmp/uv-install.sh \
35 + && rm -f /tmp/uv-install.sh \
36 + && uv --version
37
38 # This timeout (in seconds) is necessary when installing some dependencies via uv since it's likely to time out
39 # Reference: https://github.com/astral-sh/uv/pull/1694