2020-09-02 13:03:35 -07:00
|
|
|
ARG GPU
|
2020-10-12 14:22:51 -07:00
|
|
|
FROM rayproject/ray:nightly"$GPU"
|
2020-09-02 13:03:35 -07:00
|
|
|
|
|
|
|
|
# We have to uninstall wrapt this way for Tensorflow compatibility
|
|
|
|
|
COPY requirements.txt ./
|
|
|
|
|
COPY requirements_ml_docker.txt ./
|
|
|
|
|
COPY requirements_rllib.txt ./
|
|
|
|
|
COPY requirements_tune.txt ./
|
|
|
|
|
|
2020-11-12 08:41:50 -08:00
|
|
|
RUN sudo apt-get update \
|
|
|
|
|
&& sudo apt-get install -y gcc \
|
2020-09-02 13:03:35 -07:00
|
|
|
cmake \
|
|
|
|
|
libgtk2.0-dev \
|
|
|
|
|
zlib1g-dev \
|
|
|
|
|
libgl1-mesa-dev \
|
2020-12-10 01:12:46 -08:00
|
|
|
&& $HOME/anaconda3/bin/pip --use-deprecated=legacy-resolver --no-cache-dir install -r requirements.txt \
|
|
|
|
|
&& $HOME/anaconda3/bin/pip --use-deprecated=legacy-resolver --no-cache-dir install -r requirements_ml_docker.txt \
|
2020-11-30 14:12:57 -08:00
|
|
|
# Remove dataclasses & typing because they are included in Py3.7
|
|
|
|
|
&& $HOME/anaconda3/bin/pip uninstall dataclasses typing -y \
|
2020-11-12 08:41:50 -08:00
|
|
|
&& sudo rm requirements.txt && sudo rm requirements_ml_docker.txt \
|
|
|
|
|
&& sudo apt-get clean
|
2020-09-02 13:03:35 -07:00
|
|
|
|
