20 lines
622 B
Docker
20 lines
622 B
Docker
# syntax=docker/dockerfile:1.3-labs
|
|
|
|
ARG BASE_IMAGE
|
|
ARG FULL_BASE_IMAGE=rayproject/ray:nightly"$BASE_IMAGE"
|
|
FROM "$FULL_BASE_IMAGE"
|
|
|
|
COPY python/*requirements.txt \
|
|
python/requirements/ml/*requirements.txt \
|
|
python/requirements/docker/*requirements.txt ./
|
|
COPY docker/ray-ml/install-ml-docker-requirements.sh ./
|
|
|
|
RUN sudo chmod +x install-ml-docker-requirements.sh \
|
|
&& ./install-ml-docker-requirements.sh
|
|
|
|
# Export installed packages
|
|
RUN $HOME/anaconda3/bin/pip freeze > /home/ray/pip-freeze.txt
|
|
|
|
# Make sure tfp is installed correctly and matches tf version.
|
|
RUN python -c "import tensorflow_probability"
|