Update Dockerfile

Dockerfile

 ARG cuda_version=10.0
 ARG cudnn_version=7
-FROM nvidia/cuda:${cuda_version}-cudnn${cudnn_version}-devel
+# 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
 
@@ -9,19 +10,19 @@ RUN apt-get update && apt-get install -y apt-transport-https ca-certificates cur
 # RUN apt-get update && apt-get install -y docker-ce
 
 # 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