Check if GPU idx is not space or \n

alex-petrenko · Jan 16, 2023 · 7509cc4 · 7509cc4
1 parent 9e508a6
commit 7509cc4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sample_factory/utils/gpu_utils.py b/sample_factory/utils/gpu_utils.py
@@ -24,7 +24,7 @@ def get_available_gpus() -> List[int]:
     Returns indices of GPUs specified by CUDA_VISIBLE_DEVICES.
     """
     orig_visible_devices = os.environ[f"{CUDA_ENVVAR}"]
-    available_gpus = [int(g.strip()) for g in orig_visible_devices.split(",") if g]
+    available_gpus = [int(g.strip()) for g in orig_visible_devices.split(",") if g and not g.isspace()]
     return available_gpus