ray/docker/ray-deps/Dockerfile

21 lines
757 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 \
"six==1.13.0" \
"boto3==1.4.8" \
"google-api-python-client==1.7.8" \
"google-oauth" \
"kubernetes" \
"azure-cli-core==2.22.0" \
"azure-mgmt-compute==14.0.0" \
"azure-mgmt-msi==1.0.0" \
"azure-mgmt-network==10.2.0"; fi) \
&& $HOME/anaconda3/bin/pip uninstall ray -y && sudo rm $(basename $WHEEL_PATH)