new dockerfile with cuda 10.1 and cudnn7

docker/Dockerfile

@@ -1,28 +1,28 @@
-FROM ubuntu:16.04
-
-RUN apt-get update && apt-get install -y --no-install-recommends apt-utils ca-certificates apt-transport-https gnupg-curl && \
-    rm -rf /var/lib/apt/lists/* && \
-    NVIDIA_GPGKEY_SUM=d1be581509378368edeec8c1eb2958702feedf3bc3d17011adbf24efacce4ab5 && \
-    NVIDIA_GPGKEY_FPR=ae09fe4bbd223a84b2ccfce3f60f4b3d7fa2af80 && \
-    apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub && \
-    apt-key adv --export --no-emit-version -a $NVIDIA_GPGKEY_FPR | tail -n +5 > cudasign.pub && \
-    echo "$NVIDIA_GPGKEY_SUM  cudasign.pub" | sha256sum -c --strict - && rm cudasign.pub && \
-    echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/cuda.list && \
-    echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list
-
-ENV CUDA_VERSION 9.0.176
-ENV NCCL_VERSION 2.4.2
-ENV CUDA_PKG_VERSION 9-0=$CUDA_VERSION-1
-ENV CUDNN_VERSION 7.4.2.24
+FROM ubuntu:18.04
+
+RUN apt-get update && apt-get install -y --no-install-recommends apt-utils ca-certificates gnupg2 curl && \
+    curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub | apt-key add - && \
+    echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list && \
+    echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list &&\
+    apt-get purge --autoremove -y curl && \
+rm -rf /var/lib/apt/lists/*
+
+ENV CUDA_VERSION 10.1.243
+ENV NCCL_VERSION 2.4.8
+ENV CUDA_PKG_VERSION 10-1=$CUDA_VERSION-1
+ENV CUDNN_VERSION 7.6.5.32
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
-        cuda-cudart-$CUDA_PKG_VERSION && \
-    ln -s cuda-9.0 /usr/local/cuda && \
+        cuda-cudart-$CUDA_PKG_VERSION \
+        cuda-compat-10-1 && \
+    ln -s cuda-10.1 /usr/local/cuda && \
     rm -rf /var/lib/apt/lists/*
 
 RUN apt-get update && apt-get install -y --allow-unauthenticated --no-install-recommends \
         cuda-libraries-$CUDA_PKG_VERSION \
-        libnccl2=$NCCL_VERSION-1+cuda9.0 && \
+        cuda-nvtx-$CUDA_PKG_VERSION \
+        libcublas10=10.2.1.243-1 \
+        libnccl2=$NCCL_VERSION-1+cuda10.1 && \
     apt-mark hold libnccl2 && \
     rm -rf /var/lib/apt/lists/*
 
@@ -31,11 +31,19 @@ RUN apt-get update && apt-get install -y --allow-unauthenticated --no-install-re
         cuda-nvml-dev-$CUDA_PKG_VERSION \
         cuda-minimal-build-$CUDA_PKG_VERSION \
         cuda-command-line-tools-$CUDA_PKG_VERSION \
-        cuda-core-9-0=9.0.176.3-1 \
-        cuda-cublas-dev-9-0=9.0.176.4-1 \
-        libnccl-dev=$NCCL_VERSION-1+cuda9.0 && \
+        libnccl-dev=$NCCL_VERSION-1+cuda10.1 \
+        libcublas-dev=10.2.1.243-1 \
+        && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    libcudnn7=$CUDNN_VERSION-1+cuda10.1 \
+    libcudnn7-dev=$CUDNN_VERSION-1+cuda10.1 \
+    && \
+    apt-mark hold libcudnn7 && \
     rm -rf /var/lib/apt/lists/*
 
+
 ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs
 
 # NVIDIA docker 1.0.
@@ -51,7 +59,7 @@ ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
 # NVIDIA container runtime.
 ENV NVIDIA_VISIBLE_DEVICES all
 ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
-ENV NVIDIA_REQUIRE_CUDA "cuda>=9.0"
+ENV NVIDIA_REQUIRE_CUDA "cuda>=10.0 brand=tesla,driver>=384,driver<385 brand=tesla,driver>=410,driver<411"
 
 # PyTorch (Geometric) installation
 RUN rm /etc/apt/sources.list.d/cuda.list && \
@@ -97,12 +105,11 @@ ENV CONDA_DEFAULT_ENV=py36
 ENV CONDA_PREFIX=/home/user/miniconda/envs/$CONDA_DEFAULT_ENV
 ENV PATH=$CONDA_PREFIX/bin:$PATH
 
-# CUDA 9.0-specific steps.
+# CUDA 10.0-specific steps.
 RUN conda install -y -c pytorch \
-    cuda90=1.0 \
-    magma-cuda90=2.4.0 \
-    "pytorch=1.1.0=py3.6_cuda9.0.176_cudnn7.5.1_0" \
-    torchvision=0.2.1 \
+    cudatoolkit=10.1 \
+    "pytorch=1.4.0=py3.6_cuda10.1.243_cudnn7.6.3_0" \
+    torchvision=0.5.0=py36_cu101 \
  && conda clean -ya
 
 # Install HDF5 Python bindings.
@@ -118,9 +125,8 @@ RUN conda install -y requests=2.19.1 \
  && conda clean -ya
 
 # Install Graphviz.
-RUN conda install -y graphviz=2.38.0 \
+RUN conda install -y graphviz=2.40.1 python-graphviz=0.8.4 \
  && conda clean -ya
-RUN pip install graphviz==0.8.4
 
 # Install OpenCV3 Python bindings.
 RUN sudo apt-get update && sudo apt-get install -y --no-install-recommends \