fix ray install (axolotl-ai-cloud#1630)

lhl · May 16, 2024 · 891ae8a · 891ae8a
1 parent 0c49ecc
commit 891ae8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/Dockerfile-cloud-no-tmux b/docker/Dockerfile-cloud-no-tmux
@@ -16,7 +16,7 @@ COPY scripts/motd /etc/motd
 RUN pip install jupyterlab notebook ipywidgets && \
     jupyter lab clean
 RUN apt install --yes --no-install-recommends openssh-server tmux sudo && \
-    pip3 install -U --no-cache-dir grpcio ray==2.9.3 && \
+    pip3 install -U --no-cache-dir grpcio ray[default]==2.9.3 && \
     mkdir -p ~/.ssh && \
     chmod 700 ~/.ssh && \
     printf "[ ! -z \"\$TERM\" -a -r /etc/motd ] && cat /etc/motd\n" >> ~/.bashrc && \