aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7')
-rw-r--r--tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn72
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7 b/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7
index 30bc2d2806..056b4755f4 100644
--- a/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7
+++ b/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7
@@ -37,6 +37,8 @@ RUN pip --no-cache-dir install --upgrade \
RUN pip --no-cache-dir install \
ipykernel \
jupyter \
+ keras_applications==1.0.4 \
+ keras_preprocessing==1.0.2 \
matplotlib \
numpy \
scipy \