Update Dockerfile

Dockerfile

 ARG cuda_version=10.0
 ARG cudnn_version=7
-# FROM nvidia/cuda:${cuda_version}-cudnn${cudnn_version}-devel
-
-FROM docker:latest
+FROM nvidia/cuda:${cuda_version}-cudnn${cudnn_version}-devel
 
 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