Allow GPU to optionally be used

FNNDSC · Jun 3, 2024 · 45d4779 · 45d4779
1 parent e36433e
commit 45d4779
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/emerald/__init__.py b/emerald/__init__.py
@@ -1,4 +1,4 @@
-__version__ = '0.2.2'
+__version__ = '0.2.3'
 
 DISPLAY_TITLE = r"""
        _                                     _     _ 

diff --git a/emerald/__main__.py b/emerald/__main__.py
@@ -41,7 +41,8 @@
     category='MRI',
     min_memory_limit='2Gi',      # supported units: Mi, Gi
     min_cpu_limit='1000m',       # millicores, e.g. "1000m" = 1 CPU core
-    min_gpu_limit=0              # set min_gpu_limit=1 to enable GPU
+    min_gpu_limit=0,             # set min_gpu_limit=1 to enable GPU
+    max_gpu_limit=1
 )
 def main(options: Namespace, inputdir: Path, outputdir: Path):
     print(DISPLAY_TITLE, flush=True)