diff --git a/pilot/utils.py b/pilot/utils.py index 0179d12c2..607b83251 100644 --- 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