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()