Skip to content

Commit 47a0b61

Browse files
authored
Add ray[default] to wget to run distributed inference out of box (#11265)
Signed-off-by: Jiaxin Shan <seedjeffwan@gmail.com>
1 parent 5d2248d commit 47a0b61

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

Dockerfile

+1-1
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ RUN PYTHON_VERSION_STR=$(echo ${PYTHON_VERSION} | sed 's/\.//g') && \
163163
RUN echo 'tzdata tzdata/Areas select America' | debconf-set-selections \
164164
&& echo 'tzdata tzdata/Zones/America select Los_Angeles' | debconf-set-selections \
165165
&& apt-get update -y \
166-
&& apt-get install -y ccache software-properties-common git curl sudo vim python3-pip \
166+
&& apt-get install -y ccache software-properties-common git curl wget sudo vim python3-pip \
167167
&& apt-get install -y ffmpeg libsm6 libxext6 libgl1 \
168168
&& add-apt-repository ppa:deadsnakes/ppa \
169169
&& apt-get update -y \

requirements-cuda.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
-r requirements-common.txt
33

44
# Dependencies for NVIDIA GPUs
5-
ray >= 2.9
5+
ray[default] >= 2.9
66
nvidia-ml-py >= 12.560.30 # for pynvml package
77
torch == 2.5.1
88
# These must be updated alongside torch

0 commit comments

Comments
 (0)