diff --git a/dockerfiles/Dockerfile.ray-worker-intel b/dockerfiles/Dockerfile.ray-worker-intel index 7a762c6..a5c9942 100644 --- a/dockerfiles/Dockerfile.ray-worker-intel +++ b/dockerfiles/Dockerfile.ray-worker-intel @@ -58,8 +58,10 @@ USER ray ENV PATH="/home/ray/.local/bin:${PATH}" # Install Intel Extension for PyTorch (IPEX) with XPU support (uv is 10-100x faster) +# Use unsafe-best-match to get torch from PyPI and IPEX from Intel index RUN --mount=type=cache,target=/home/ray/.cache/uv,uid=1000,gid=1000 \ uv pip install --system \ + --index-strategy unsafe-best-match \ torch==2.5.1 \ intel-extension-for-pytorch==2.5.10+xpu \ --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/