From 46740958f28b504221ed8028585a94ea93cc7aa3 Mon Sep 17 00:00:00 2001
From: Yikun Jiang <yikunkero@gmail.com>
Date: Fri, 28 Feb 2025 15:23:18 +0800
Subject: [PATCH] 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>
---
 Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 9b4c075..0854198 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -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"]