1 files changed, 10 insertions, 0 deletions
diff --git a/utils/misc.py b/utils/misc.py
new file mode 100644
index 0000000..b850830
--- /dev/null
+++ b/utils/misc.py
@@ -0,0 +1,10 @@
+import os
+
+from utils.configuration import SystemConfiguration
+
+
+def set_visible_cuda(config: SystemConfiguration):
+    """Set environment variable CUDA device(s)"""
+    CUDA_VISIBLE_DEVICES = config.get('CUDA_VISIBLE_DEVICES', None)
+    if CUDA_VISIBLE_DEVICES:
+        os.environ['CUDA_VISIBLE_DEVICES'] = CUDA_VISIBLE_DEVICES