mirror of
https://github.com/vale981/ray
synced 2025-03-06 02:21:39 -05:00
20 lines
732 B
Docker
20 lines
732 B
Docker
ARG GPU=""
|
|
FROM rayproject/base-deps:nightly"$GPU"
|
|
# If this arg is not "autoscaler" then no autoscaler requirements will be included
|
|
ARG AUTOSCALER="autoscaler"
|
|
ARG WHEEL_PATH
|
|
# For Click
|
|
ENV LC_ALL=C.UTF-8
|
|
ENV LANG=C.UTF-8
|
|
COPY $WHEEL_PATH .
|
|
RUN $HOME/anaconda3/bin/pip --no-cache-dir install $(basename $WHEEL_PATH)[all] \
|
|
$(if [ "$AUTOSCALER" = "autoscaler" ]; then echo \
|
|
"boto3==1.4.8" \
|
|
"google-api-python-client==1.7.8" \
|
|
"google-oauth" \
|
|
"kubernetes" \
|
|
"azure-cli-core==2.4.0" \
|
|
"azure-mgmt-compute==12.0.0" \
|
|
"azure-mgmt-msi==1.0.0" \
|
|
"azure-mgmt-network==10.1.0"; fi) \
|
|
&& $HOME/anaconda3/bin/pip uninstall ray -y && sudo rm $(basename $WHEEL_PATH)
|