diff --git a/docker/examples/Dockerfile b/docker/examples/Dockerfile index 6883c5a64..04df50a82 100644 --- a/docker/examples/Dockerfile +++ b/docker/examples/Dockerfile @@ -5,6 +5,8 @@ FROM ray-project/deploy # This updates numpy to 1.14 and mutes errors from other libraries RUN conda install -y numpy 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] opencv-python-headless tensorflow lz4 keras pytest-timeout smart_open RUN pip install -U h5py # Mutes FutureWarnings RUN pip install --upgrade bayesian-optimization diff --git a/docker/tune_test/Dockerfile b/docker/tune_test/Dockerfile index 6e098d521..41ef63390 100644 --- a/docker/tune_test/Dockerfile +++ b/docker/tune_test/Dockerfile @@ -7,6 +7,8 @@ FROM ray-project/base-deps RUN pip install -U https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-0.8.0.dev1-cp27-cp27mu-manylinux1_x86_64.whl 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 RUN pip install --upgrade bayesian-optimization RUN pip install --upgrade git+git://github.com/hyperopt/hyperopt.git