diff --git a/Dockerfile b/Dockerfile index 4762801a0b2cfb91956010ccf10717f56ceb6257..8e5d0403b91e6ce83ba0eedf243c93426b2b61d4 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -ARG cuda_version=9.0 +ARG cuda_version=1.0 ARG cudnn_version=7 FROM nvidia/cuda:${cuda_version}-cudnn${cudnn_version}-devel