diff --git a/Dockerfile b/Dockerfile
index de85a59780384e552259e14bd43e3f6543b8a4f1..111df954b9d4b3dd95e2ed8afc0392825aedd83a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,32 +1,26 @@
 ARG cuda_version=10.0
 ARG cudnn_version=7
-# FROM nvidia/cuda:${cuda_version}-cudnn${cudnn_version}-devel
-
-FROM docker:latest
-RUN apt-get update && apt-get install -y apt-transport-https ca-certificates curl software-properties-common
-# RUN curl -fsSL https://download.docker.com/linux/ubuntu/gpg
-
-# RUN apt-get update && add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu bionic test"
-# RUN apt-get update && apt-get install -y docker-ce
+FROM nvidia/cuda:${cuda_version}-cudnn${cudnn_version}-devel
 
 # Install system packages
-# RUN apt-get update && apt-get install -y --no-install-recommends \
-#       bzip2 \
-#       g++ \
-#       git \
-#       graphviz \
-#       libgl1-mesa-glx \
-#       libhdf5-dev \
-#       sudo\
-#       openmpi-bin \
-#       xvfb \
-#       screen \
-#       wget && \
-#     rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y --no-install-recommends \
+      bzip2 \
+      g++ \
+      git \
+      graphviz \
+      libgl1-mesa-glx \
+      libhdf5-dev \
+      sudo\
+      openmpi-bin \
+      xvfb \
+      screen \
+      wget && \
+    rm -rf /var/lib/apt/lists/*
 
 # Install conda
 ENV CONDA_DIR /opt/conda
 ENV PATH $CONDA_DIR/bin:$PATH
+
 RUN wget --quiet --no-check-certificate https://repo.continuum.io/miniconda/Miniconda3-4.2.12-Linux-x86_64.sh && \
     echo "c59b3dd3cad550ac7596e0d599b91e75d88826db132e4146030ef471bb434e9a *Miniconda3-4.2.12-Linux-x86_64.sh" | sha256sum -c - && \
     /bin/bash /Miniconda3-4.2.12-Linux-x86_64.sh -f -b -p $CONDA_DIR && \
@@ -39,11 +33,10 @@ ENV NB_UID 1000
 
 RUN useradd -m -s /bin/bash -N -u $NB_UID $NB_USER && \
     chown $NB_USER $CONDA_DIR -R && \
+    chown $NB_USER /userdata/kerasData -R && \
     mkdir -p / && \
     chown $NB_USER / && \
     chpasswd $NB_USER:ar-noc
-RUN usermod -aG docker ${NB_USER}
-
 
 USER $NB_USER
 
@@ -78,6 +71,7 @@ RUN conda install -y python=${python_version} && \
     conda clean -yt
     # pip install git+git://github.com/keras-team/keras.git && \
 
+
 #ADD theanorc /home/keras/.theanorc
 
 ENV LC_ALL=C.UTF-8