sparkleman
commited on
Commit
·
d055fef
1
Parent(s):
9b15eee
UPDATE: Torch memory
Browse files
app.py
CHANGED
|
@@ -39,7 +39,7 @@ def logGPUState():
|
|
| 39 |
if "cuda" in CONFIG.STRATEGY:
|
| 40 |
gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
|
| 41 |
logger.info(
|
| 42 |
-
f"[STATUS] vram {format_bytes(gpu_info.total)} used {format_bytes(gpu_info.used)} free {format_bytes(gpu_info.free)}"
|
| 43 |
)
|
| 44 |
|
| 45 |
|
|
|
|
| 39 |
if "cuda" in CONFIG.STRATEGY:
|
| 40 |
gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
|
| 41 |
logger.info(
|
| 42 |
+
f"[STATUS] Torch - {format_bytes(torch.cuda.memory_allocated())} - NVML - vram {format_bytes(gpu_info.total)} used {format_bytes(gpu_info.used)} free {format_bytes(gpu_info.free)}"
|
| 43 |
)
|
| 44 |
|
| 45 |
|