forked from ray-project/ray
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
37 lines (35 loc) · 915 Bytes
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# The examples Docker image adds dependencies needed to run the examples
FROM rayproject/ray:latest
# Needed to run Tune example with a 'plot' call - which does not actually render a plot, but throws an error.
RUN apt-get update && apt-get install -y zlib1g-dev libgl1-mesa-dev libgtk2.0-dev && apt-get clean
RUN pip install --no-cache-dir -U pip \
autorom[accept-rom-license] \
gym[atari] \
scikit-image \
tensorflow \
lz4 \
pytest-timeout \
smart_open \
tensorflow_probability \
dm_tree \
h5py # Mutes FutureWarnings \
bayesian-optimization \
hyperopt \
ConfigSpace==0.4.10 \
sigopt \
nevergrad \
scikit-optimize \
hpbandster \
lightgbm \
xgboost \
torch \
torchvision \
tensorboardX \
dragonfly-opt \
zoopt \
tabulate \
mlflow \
pytest-remotedata>=0.3.1 \
matplotlib \
jupyter \
pandas