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 <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
authored and GitHub committed 2022-12-01 13:32:55 -08:00
1 parent 185d475d93
commit 1ce464f689
1 file changed
+5
+5
View File
@@ -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: