mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
30 lines
1.3 KiB
Docker
30 lines
1.3 KiB
Docker
# The stress_test Docker image build a self-contained Ray instance for launching Ray.
|
|
|
|
FROM ray-project/base-deps
|
|
|
|
# We install ray and boto3 to enable the ray autoscaler as
|
|
# a test runner.
|
|
RUN conda install -y numpy
|
|
RUN pip install -U pip
|
|
RUN pip install -U https://ray-wheels.s3-us-west-2.amazonaws.com/latest/ray-0.8.0.dev6-cp36-cp36m-manylinux1_x86_64.whl || pip install -U https://ray-wheels.s3-us-west-2.amazonaws.com/latest/ray-0.8.0.dev5-cp36-cp36m-manylinux1_x86_64.whl
|
|
RUN pip install -U boto3
|
|
# We install this after the latest wheels -- this should not override the latest wheels.
|
|
RUN apt-get install -y zlib1g-dev
|
|
# The following is needed to support TensorFlow 1.14
|
|
RUN conda remove -y --force wrapt
|
|
RUN pip install gym[atari]==0.10.11 opencv-python-headless tensorflow lz4 keras pytest-timeout smart_open torch torchvision
|
|
RUN pip install --upgrade bayesian-optimization
|
|
RUN pip install --upgrade hyperopt==0.1.2
|
|
RUN pip install ConfigSpace==0.4.10
|
|
RUN pip install --upgrade sigopt nevergrad scikit-optimize hpbandster lightgbm xgboost
|
|
RUN pip install -U mlflow
|
|
RUN pip install -U pytest-remotedata>=0.3.1
|
|
|
|
# RUN mkdir -p /root/.ssh/
|
|
|
|
# We port the source code in so that we run the most up-to-date stress tests.
|
|
ADD ray.tar /ray
|
|
ADD git-rev /ray/git-rev
|
|
RUN python /ray/python/ray/setup-dev.py --yes
|
|
|
|
WORKDIR /ray
|