Update utils.py

get_gpu_memory should return after the loop is over
2025-08-10 20:52:33 +00:00 · 2023-05-23 14:24:00 +08:00 · 2023-05-23 14:24:00 +08:00 · a1b4c92f6b
commit a1b4c92f6b
parent e847a3fc7a
1 changed files with 1 additions and 1 deletions
--- a/pilot/utils.py
+++ b/pilot/utils.py
@ -33,7 +33,7 @@ def get_gpu_memory(max_gpus=None):
            allocated_memory = torch.cuda.memory_allocated() / (1024 ** 3)
            available_memory = total_memory - allocated_memory
            gpu_memory.append(available_memory)
-        return gpu_memory
+    return gpu_memory