diff options
Diffstat (limited to 'tensorflow/tools/docker/Dockerfile.gpu')
-rw-r--r-- | tensorflow/tools/docker/Dockerfile.gpu | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.gpu b/tensorflow/tools/docker/Dockerfile.gpu index 5ec1e60f00..aa0e0face1 100644 --- a/tensorflow/tools/docker/Dockerfile.gpu +++ b/tensorflow/tools/docker/Dockerfile.gpu @@ -37,6 +37,8 @@ RUN pip --no-cache-dir install \ h5py \ ipykernel \ jupyter \ + keras_applications==1.0.4 \ + keras_preprocessing==1.0.2 \ matplotlib \ numpy==1.14.5 \ pandas \ |