From 3df4aa83d3abcb8ff40601907c298eed71073330 Mon Sep 17 00:00:00 2001 From: Byungheong Jeong <byungheon.jeong@gmail.com> Date: Wed, 3 Jun 2020 23:03:34 -0700 Subject: [PATCH] Update Dockerfile --- Dockerfile | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index 2095c21..9e86048 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 -- GitLab