Skip to content

Commit

Permalink
[image] fix llm image base dependencies (#50612)
Browse files Browse the repository at this point in the history
strict check on script execution, and properly install the dependencies

Signed-off-by: Lonnie Liu <lonnie@anyscale.com>
  • Loading branch information
aslonnie authored and israbbani committed Feb 25, 2025
1 parent 443d3a6 commit 8f3c745
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion docker/ray-llm/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -8,14 +8,16 @@ COPY python/requirements_*.txt ./
RUN <<EOF
#!/bin/bash

set -euo pipefail

PYTHON_CODE="$(python -c "import sys; v=sys.version_info; print(f'py{v.major}{v.minor}')")"

if [[ "${PYTHON_CODE}" != "py311" ]]; then
echo "ray-llm only support Python 3.11 now (this image is for ${PYTHON_CODE})."
exit 1
fi

pip install --no-deps -r "python/requirements_compiled_rayllm_py${PYTHON_CODE}.txt"
pip install --no-deps -r "requirements_compiled_rayllm_${PYTHON_CODE}.txt"

# Export installed packages
$HOME/anaconda3/bin/pip freeze > /home/ray/pip-freeze.txt
Expand Down

0 comments on commit 8f3c745

Please sign in to comment.