p04798526
/
LLaMA-Factory-Mirror
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

change: gpu status

This commit is contained in:
wql 2024-08-20 17:38:32 +08:00
parent 8793d13920
commit 6e99c064ad
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/gpu_status.py
View File

@ -9,7 +9,7 @@ def main():
gpuDeviceCount = pynvml.nvmlDeviceGetCount() gpuDeviceCount = pynvml.nvmlDeviceGetCount()
start_time = time.time() start_time = time.time()
while time.time() - start_time < 3600 *16: while time.time() - start_time < 3600 *24:
# print(time.time() - start_time) # print(time.time() - start_time)
all_gpu_status = [] all_gpu_status = []
for i in range(gpuDeviceCount): for i in range(gpuDeviceCount):
@ -28,7 +28,7 @@ def main():
) )
with open("./results/gpu_status.json", "a", encoding="utf-8") as f: with open("./results/gpu_status.json", "a", encoding="utf-8") as f:
f.write(json.dumps(logs) + "\n") f.write(json.dumps(logs) + "\n")
# print(logs) print(logs)
time.sleep(60) time.sleep(60)
pynvml.nvmlShutdown() pynvml.nvmlShutdown()