Fix tensorflow-1.14 installation in jenkins (#5007)

ray-project · Jun 21, 2019 · 2e342ef · 2e342ef
1 parent a7f84b5
commit 2e342ef
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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