ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-01-31 09:06:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Changed default docker image from nvidia/cuda to "nvidia/cuda:10.1-runtime-ubuntu18.04" to support cudnn frameworks (TF)

Browse Source
This commit is contained in:
allegroai 2020-07-02 01:35:57 +03:00
parent f693fa165c
commit 1f4d358316
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/trains.conf
View File

@ -106,7 +106,7 @@ agent {
default_docker: {
# default docker image to use when running in docker mode
image: "nvidia/cuda"
image: "nvidia/cuda:10.1-runtime-ubuntu18.04"
# optional arguments to pass to docker image
# arguments: ["--ipc=host"]

2
trains_agent/backend_api/config/default/agent.conf
View File

@ -92,7 +92,7 @@
default_docker: {
# default docker image to use when running in docker mode
image: "nvidia/cuda:10.2-runtime-ubuntu18.04"
image: "nvidia/cuda:10.1-runtime-ubuntu18.04"
# optional arguments to pass to docker image
# arguments: ["--ipc=host", ]