github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-09 12:18:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update utils.py (#82)

Browse Source 
get_gpu_memory should return after the loop is over
This commit is contained in:
magic.chen 2023-05-23 16:02:13 +08:00 committed by GitHub
commit 528c0e78f6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/utils.py
View File

@ -33,7 +33,7 @@ def get_gpu_memory(max_gpus=None):
allocated_memory = torch.cuda.memory_allocated() / (1024 ** 3) allocated_memory = torch.cuda.memory_allocated() / (1024 ** 3)
available_memory = total_memory - allocated_memory available_memory = total_memory - allocated_memory
gpu_memory.append(available_memory) gpu_memory.append(available_memory)
return gpu_memory return gpu_memory