Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -77,7 +77,8 @@ def evaluate(
|
|
77 |
out_last = i + 1
|
78 |
|
79 |
gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
|
80 |
-
|
|
|
81 |
del out
|
82 |
del state
|
83 |
gc.collect()
|
|
|
77 |
out_last = i + 1
|
78 |
|
79 |
gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
|
80 |
+
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
81 |
+
print(f'{timestamp} - vram {gpu_info.total} used {gpu_info.used} free {gpu_info.free}')
|
82 |
del out
|
83 |
del state
|
84 |
gc.collect()
|