Add ray to docker image (#197)

### What this PR does / why we need it?
Add ray to docker image to make `ray` work

### Does this PR introduce _any_ user-facing change?
No

### How was this patch tested?
CI passed

Signed-off-by: Yikun Jiang <yikunkero@gmail.com>
This commit is contained in:
Yikun Jiang
2025-02-28 15:23:18 +08:00
committed by GitHub
parent 81dfaae88b
commit 46740958f2

View File

@@ -41,7 +41,7 @@ RUN VLLM_TARGET_DEVICE="empty" python3 -m pip install /workspace/vllm/
# Install vllm-ascend main
RUN python3 -m pip install /workspace/vllm-ascend/ --extra-index https://download.pytorch.org/whl/cpu/
# Install modelscope
RUN python3 -m pip install modelscope
# Install modelscope (for fast download) and ray/net-tools (for multinode)
RUN python3 -m pip install modelscope ray net-tools
CMD ["/bin/bash"]