From 24cdefb8c5b472f29dd115d26b31188cd214540e Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Sun, 20 Mar 2022 23:20:19 +0200
Subject: [PATCH] Fix docstring

---
 clearml/automation/controller.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/clearml/automation/controller.py b/clearml/automation/controller.py
index 12424add..4af50821 100644
--- a/clearml/automation/controller.py
+++ b/clearml/automation/controller.py
@@ -291,7 +291,8 @@ class PipelineController(object):
             - match git repository branch to a previous step
                 task_overrides={'script.branch': '${stage1.script.branch}', 'script.version_num': ''}
             - change container image
-                task_overrides={'container.image': '${stage1.container.image}'}
+                task_overrides={'container.image': 'nvidia/cuda:11.6.0-devel-ubuntu20.04',
+                                'container.arguments': '--ipc=host'}
             - match container image to a previous step
                 task_overrides={'container.image': '${stage1.container.image}'}
         :param execution_queue: Optional, the queue to use for executing this specific step.