apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): show VRAM percent in logs

This commit is contained in:
Sean Sube 2023-03-05 19:23:23 -06:00
parent 7a3a81a4ef
commit 39b9741b24
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/utils.py
View File

@ -97,10 +97,12 @@ def run_gc(devices: Optional[List[DeviceParams]] = None):
torch.cuda.empty_cache()
torch.cuda.ipc_collect()
mem_free, mem_total = torch.cuda.mem_get_info()
mem_pct = (1 - (mem_free / mem_total)) * 100
logger.debug(
"remaining CUDA VRAM usage: %s of %s",
"CUDA VRAM usage: %s of %s (%.2f%%)",
(mem_total - mem_free),
mem_total,
mem_pct,
)

4
api/onnx_web/worker/pool.py
View File

@ -291,7 +291,9 @@ class DevicePoolExecutor:
logger.debug("shutting down worker for device %s", device)
worker.join(self.join_timeout)
if worker.is_alive():
logger.error("leaking worker for device %s could not be shut down", device)
logger.error(
"leaking worker for device %s could not be shut down", device
)
self.leaking[:] = [dw for dw in self.leaking if dw[1].is_alive()]