gpu-libs installer instead of splitted tensorflow and pytorch

.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 && \

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 "

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 && \