Small fix when getting a list of GPUs

alex-petrenko · Jan 16, 2023 · 9e508a6 · 9e508a6
1 parent ceef583
commit 9e508a6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/docs/11-release-notes/release-notes.md b/docs/11-release-notes/release-notes.md
@@ -5,8 +5,9 @@
 * Added cfg parameters `--lr_adaptive_min` and `--lr_adaptive_max` to control the minimum and maximum adaptive learning rate
 * Added Brax environment support + custom brax renderer for enjoy scripts
 * Automatically set `--recurrence` based on feed-forward vs RNN training
-* Added `--enjoy_script` and `--train_script` for generating the model card when uploading to the Hugging Face Hub
+* Added `--enjoy_script` and `--train_script` for generating the model card when uploading to the Hugging Face Hub (thank you Andrew!)
 * Fixed video name when generating Hugging Face model card
+* Fixed small DMLab-related bug (thank you Lucy!)
 
 ##### v2.0.2
 

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) for g in orig_visible_devices.split(",") if g]
+    available_gpus = [int(g.strip()) for g in orig_visible_devices.split(",") if g]
     return available_gpus