Support GPUs in Singularity runtime too

hubmapconsortium · Aug 27, 2021 · 4277d62 · 4277d62
1 parent d29e578
commit 4277d62
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cwltool/singularity.py b/cwltool/singularity.py
@@ -388,6 +388,8 @@ def create_runtime(
             "--ipc",
             "--cleanenv",
         ]
+        if getattr(self, 'docker_gpu_flag', False):
+            runtime.append('--nv')
         if _singularity_supports_userns():
             runtime.append("--userns")
         else: