mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
21 lines
656 B
Docker
21 lines
656 B
Docker
ARG GPU
|
|
FROM rayproject/ray:latest"$GPU"
|
|
|
|
# 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 ./
|
|
|
|
RUN apt-get update \
|
|
&& apt-get install -y gcc \
|
|
cmake \
|
|
libgtk2.0-dev \
|
|
zlib1g-dev \
|
|
libgl1-mesa-dev \
|
|
&& $HOME/anaconda3/bin/pip --no-cache-dir install -r requirements.txt \
|
|
&& $HOME/anaconda3/bin/pip --no-cache-dir install -r requirements_ml_docker.txt \
|
|
&& rm requirements.txt && rm requirements_ml_docker.txt \
|
|
&& apt-get remove cmake gcc -y \
|
|
&& apt-get clean
|
|
|