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

Merge remote-tracking branch 'origin/main' into store_connector

Browse Source
This commit is contained in:
aries-ckt 2023-05-23 22:04:03 +08:00
commit 90b793f40e
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