diff --git a/docs/clearml.conf b/docs/clearml.conf index 95ffc97..92690e6 100644 --- a/docs/clearml.conf +++ b/docs/clearml.conf @@ -141,7 +141,7 @@ agent { default_docker: { # default docker image to use when running in docker mode - image: "nvidia/cuda:10.1-runtime-ubuntu18.04" + image: "nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04" # optional arguments to pass to docker image # arguments: ["--ipc=host"]