From 2e342ef71fd12066913d019140bc6ea43d74d6eb Mon Sep 17 00:00:00 2001 From: Philipp Moritz Date: Fri, 21 Jun 2019 11:04:40 -0700 Subject: [PATCH] Fix tensorflow-1.14 installation in jenkins (#5007) --- docker/examples/Dockerfile | 2 ++ docker/tune_test/Dockerfile | 2 ++ 2 files changed, 4 insertions(+) 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