Compare commits

...

2 Commits

Author SHA1 Message Date
wql 83c41567b3 Merge branch 'main' of https://osredm.com/p04798526/LLaMA-Factory-Mirror 2024-08-20 17:39:26 +08:00
wql 6e99c064ad change: gpu status 2024-08-20 17:38:32 +08:00
1 changed files with 2 additions and 2 deletions

View File

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