use nvidia/cuda:10.1-base-ubuntu18.04 as base image

src/Dockerfile.gpulibs

@@ -2,11 +2,6 @@ LABEL maintainer="Christoph Schranz <christoph.schranz@salzburgresearch.at>"
 
 # Install Tensorflow, check compatibility here: https://www.tensorflow.org/install/gpu
 # installation via conda leads to errors in version 4.8.2
-#RUN conda install --quiet --yes \
-#    'tensorflow-gpu=2.1*' \
-#    'keras-gpu' && \
-#    fix-permissions $CONDA_DIR && \
-#    fix-permissions /home/$NB_USER
 RUN pip install --upgrade pip && \
     pip install --no-cache-dir "tensorflow-gpu>=2.1.*" && \
     pip install --no-cache-dir keras
@@ -15,9 +10,6 @@ RUN pip install --upgrade pip && \
 RUN conda install --quiet --yes \
     pyyaml mkl mkl-include setuptools cmake cffi typing
 
-# Deep-NN libs
-RUN apt-get update && apt-get install -y libcudnn7-dev && rm -rf /var/lib/apt/lists/* 
-
 # Check compatibility here: https://pytorch.org/get-started/locally/
 RUN conda install --quiet --yes \
      pytorch \