aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/ci_build/Dockerfile.cmake
diff options
context:
space:
mode:
authorGravatar Yifei Feng <yifeif@google.com>2018-06-14 14:56:28 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-06-14 14:59:08 -0700
commitc4eafb49612a694386bbda1f51dffb6951ec9cf1 (patch)
tree552dda09acb58042c1466771aed90bf4ca027ccf /tensorflow/tools/ci_build/Dockerfile.cmake
parentf01d25471dbe26f0a1116009badc4af169f82b02 (diff)
Install Keras dependencies.
PiperOrigin-RevId: 200623983
Diffstat (limited to 'tensorflow/tools/ci_build/Dockerfile.cmake')
-rw-r--r--tensorflow/tools/ci_build/Dockerfile.cmake2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/tools/ci_build/Dockerfile.cmake b/tensorflow/tools/ci_build/Dockerfile.cmake
index d5dea4f3e4..e8c3199828 100644
--- a/tensorflow/tools/ci_build/Dockerfile.cmake
+++ b/tensorflow/tools/ci_build/Dockerfile.cmake
@@ -28,6 +28,8 @@ RUN pip install --upgrade astor
RUN pip install --upgrade gast
RUN pip install --upgrade numpy
RUN pip install --upgrade termcolor
+RUN pip install keras_applications==1.0.2
+RUN pip install keras_preprocessing==1.0.1
# Install golang
RUN apt-get install -t xenial-backports -y golang-1.9