From f8848a73b4873a65100dcc89d9fac169e89e03eb Mon Sep 17 00:00:00 2001
From: Christoph Schranz <christoph.schranz@salzburgresearch.at>
Date: Mon, 24 Feb 2020 21:35:15 +0100
Subject: [PATCH] gpu-libs installer instead of splitted tensorflow and pytorch

---
 .build/Dockerfile                              | 17 +++++------------
 generate_Dockerfile.sh                         | 10 ++--------
 src/{Dockerfile.pytorch => Dockerfile.gpulibs} |  8 ++++++++
 3 files changed, 15 insertions(+), 20 deletions(-)
 rename src/{Dockerfile.pytorch => Dockerfile.gpulibs} (71%)

diff --git a/.build/Dockerfile b/.build/Dockerfile
index c007c06..16d0e50 100755
--- a/.build/Dockerfile
+++ b/.build/Dockerfile
@@ -353,24 +353,17 @@ RUN julia -e 'import Pkg; Pkg.update()' && \
     fix-permissions $JULIA_PKGDIR $CONDA_DIR/share/jupyter
 
 ############################################################################
-################ Dependency: jupyter/tensorflow-notebook ###################
+########################## Dependency: gpulibs #############################
 ############################################################################
 
-# Copyright (c) Jupyter Development Team.
-# Distributed under the terms of the Modified BSD License.
-
-LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"
-
 # Install Tensorflow
-RUN pip install --quiet \
-    'tensorflow==2.1.0' && \
+RUN conda install --quiet --yes \
+    'tensorflow=1.13*' \
+    'keras=2.2*' && \
+    conda clean --all -f -y && \
     fix-permissions $CONDA_DIR && \
     fix-permissions /home/$NB_USER
 
-############################################################################
-########################## Dependency: pytorch #############################
-############################################################################
-
 # Install PyTorch dependencies
 RUN conda install --quiet --yes \
     pyyaml mkl mkl-include setuptools cmake cffi typing && \
diff --git a/generate_Dockerfile.sh b/generate_Dockerfile.sh
index 09865fe..f9925d7 100755
--- a/generate_Dockerfile.sh
+++ b/generate_Dockerfile.sh
@@ -53,21 +53,15 @@ echo "
 " >> $DOCKERFILE
 cat $STACKS_DIR/datascience-notebook/Dockerfile | grep -v BASE_CONTAINER >> $DOCKERFILE
 
-echo "
-############################################################################
-################ Dependency: jupyter/tensorflow-notebook ###################
-############################################################################
-" >> $DOCKERFILE
-cat $STACKS_DIR/tensorflow-notebook/Dockerfile | grep -v BASE_CONTAINER >> $DOCKERFILE
 
 # Note that the following step also installs the cudatoolkit, which is
 # essential to access the GPU.
 echo "
 ############################################################################
-########################## Dependency: pytorch #############################
+########################## Dependency: gpulibs #############################
 ############################################################################
 " >> $DOCKERFILE
-cat src/Dockerfile.pytorch >> $DOCKERFILE
+cat src/Dockerfile.gpulibs >> $DOCKERFILE
 
 
 echo "
diff --git a/src/Dockerfile.pytorch b/src/Dockerfile.gpulibs
similarity index 71%
rename from src/Dockerfile.pytorch
rename to src/Dockerfile.gpulibs
index ad344ed..7d5659a 100644
--- a/src/Dockerfile.pytorch
+++ b/src/Dockerfile.gpulibs
@@ -1,3 +1,11 @@
+# Install Tensorflow
+RUN conda install --quiet --yes \
+    'tensorflow=1.13*' \
+    'keras=2.2*' && \
+    conda clean --all -f -y && \
+    fix-permissions $CONDA_DIR && \
+    fix-permissions /home/$NB_USER
+
 # Install PyTorch dependencies
 RUN conda install --quiet --yes \
     pyyaml mkl mkl-include setuptools cmake cffi typing && \