fix
Some checks failed
Build and Push Images / determine-version (push) Successful in 4s
Build and Push Images / build (Dockerfile.ray-worker-rdna2, rdna2) (push) Has been cancelled
Build and Push Images / build (Dockerfile.ray-worker-strixhalo, strixhalo) (push) Has been cancelled
Build and Push Images / Release (push) Has been cancelled
Build and Push Images / Notify (push) Has been cancelled
Build and Push Images / build (Dockerfile.ray-worker-intel, intel) (push) Has been cancelled
Build and Push Images / build (Dockerfile.ray-worker-nvidia, nvidia) (push) Has been cancelled

This commit is contained in:
2026-02-09 11:31:18 -05:00
parent 5f2d167ba0
commit b0c58b98a0

View File

@@ -66,12 +66,16 @@ USER ray
# index — TheRock nightlies have an incompatible c10::hip ABI that causes
# undefined symbol errors in vllm._C and vllm._rocm_C.
# HSA_OVERRIDE_GFX_VERSION=11.0.0 makes gfx1151 appear as gfx1100 to this torch.
# --index-strategy unsafe-best-match: let uv pull each package from whichever
# index has the best-matching version (vllm from rocm/, torch from rocm7.0,
# setuptools/others from PyPI).
RUN --mount=type=cache,target=/home/ray/.cache/uv,uid=1000,gid=1000 \
uv pip install --system 'setuptools>=77.0.3,<81.0' && \
uv pip install --system \
--index-strategy unsafe-best-match \
--prerelease=allow \
--extra-index-url https://wheels.vllm.ai/rocm/ \
--extra-index-url https://download.pytorch.org/whl/rocm7.0 \
vllm \
'vllm==0.15.1+rocm700' \
torch torchaudio torchvision \
'transformers>=4.35.0,<5.0' \
'accelerate>=0.25.0,<1.0' \