From 6e99c064adc09e921271ab8ceda6a4c5e6a538bc Mon Sep 17 00:00:00 2001 From: wql Date: Tue, 20 Aug 2024 17:38:32 +0800 Subject: [PATCH] change: gpu status --- src/gpu_status.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/gpu_status.py b/src/gpu_status.py index 7b688519..cabff59a 100644 --- a/src/gpu_status.py +++ b/src/gpu_status.py @@ -9,7 +9,7 @@ def main(): gpuDeviceCount = pynvml.nvmlDeviceGetCount() start_time = time.time() - while time.time() - start_time < 3600 *16: + while time.time() - start_time < 3600 *24: # print(time.time() - start_time) all_gpu_status = [] for i in range(gpuDeviceCount): @@ -28,7 +28,7 @@ def main(): ) with open("./results/gpu_status.json", "a", encoding="utf-8") as f: f.write(json.dumps(logs) + "\n") - # print(logs) + print(logs) time.sleep(60) pynvml.nvmlShutdown()