diff --git a/Dockerfile b/Dockerfile
index 2095c21b9a191153a6f19c385f26e030c6bda1e8..9e860484648a1b3254514089e407f63e6c70febf 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,8 +1,6 @@
 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