From 1ce464f6890ed1afe887ab8eed78804ae5933aa8 Mon Sep 17 00:00:00 2001 From: Victor Sonck Date: Thu, 1 Dec 2022 22:32:55 +0100 Subject: [PATCH] Add docker info for ClearML remote execution (#10142) * Add docker info for ClearML remote execution * add additional clearml options to handle different python versions * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: Glenn Jocher Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> --- utils/loggers/clearml/clearml_utils.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/utils/loggers/clearml/clearml_utils.py b/utils/loggers/clearml/clearml_utils.py index fe5f597..08aa9fd 100644 --- a/utils/loggers/clearml/clearml_utils.py +++ b/utils/loggers/clearml/clearml_utils.py @@ -97,6 +97,11 @@ class ClearmlLogger: # will have to be added manually! self.task.connect(hyp, name='Hyperparameters') + # Make sure the code is easily remotely runnable by setting the docker image to use by the remote agent + self.task.set_base_docker("ultralytics/yolov5:latest", + docker_arguments='--ipc=host -e="CLEARML_AGENT_SKIP_PYTHON_ENV_INSTALL=1"', + docker_setup_bash_script='pip install clearml') + # Get ClearML Dataset Version if requested if opt.data.startswith('clearml://'): # data_dict should have the following keys: