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