refs #32: add - keras to python3-tensorflow-gpu image

lablup · Apr 14, 2017 · b56a2d4 · b56a2d4
1 parent 75fee78
commit b56a2d4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python3-tensorflow-gpu/Dockerfile b/python3-tensorflow-gpu/Dockerfile
@@ -73,6 +73,9 @@ ADD tensorflow-1.0.0-cp36-cp36m-linux_x86_64.whl /tmp
 USER work
 RUN /home/sorna/pyenv-run.sh pip install /tmp/tensorflow-*.whl
 
+# Install Keras
+RUN /home/sorna/pyenv-run.sh pip install keras
+
 # Secure installation scripts
 USER root
 ADD run.sh /home/sorna/run.sh