Update Dockerfile

Dockerfile

@@ -2,20 +2,26 @@ ARG cuda_version=10.1
 ARG cudnn_version=7
 FROM nvidia/cuda:${cuda_version}-cudnn${cudnn_version}-devel
 
-ENV NB_USER kerasTester
-ENV NB_UID 1000
+# ENV NB_USER kerasTester
+# ENV NB_UID 1000
+
+ARG NB_USER="kerasTester"
+ARG NB_UID="1000"
+ARG NB_GID="100"
+
 # RUN mkdir /userdata/kerasData
 
 RUN apt-get update && \
       apt-get -y install sudo
 
-RUN useradd -m -s /bin/bash -N -u $NB_UID $NB_USER && \
+
+# RUN useradd -m -s /bin/bash -N -u $NB_UID $NB_USER && \
     # chown $NB_USER $CONDA_DIR -R && \
     # chown $NB_USER /userdata/kerasData -R && \
     #    chown $NB_USER / -R && \
-    mkdir -p / && \
-    chpasswd $NB_USER:test && \
-    usermod -aG sudo $NB_USER
+    # mkdir -p / && \
+    # chpasswd $NB_USER:test && \
+    # usermod -aG sudo $NB_USER
 
 WORKDIR /userdata/kerasData
 
@@ -34,8 +40,15 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     rm -rf /var/lib/apt/lists/*
 
 # Install conda
-ENV CONDA_DIR /opt/conda
-ENV PATH $CONDA_DIR/bin:$PATH
+# ENV CONDA_DIR /opt/conda
+ENV CONDA_DIR=/opt/conda \
+    SHELL=/bin/bash \
+    NB_USER=$NB_USER \
+    NB_UID=$NB_UID \
+    NB_GID=$NB_GID 
+
+ENV PATH $CONDA_DIR/bin:$PATH \
+  HOME /userdata/kerasData
 
 RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-4.7.12.1-Linux-x86_64.sh -O ~/miniconda.sh && \
     /bin/bash ~/miniconda.sh -b -p /opt/conda && \