YAO/YAO-agent
1
0
Fork 0
mirror of https://github.com/newnius/YAO-agent.git synced 2025-06-07 13:51:56 +00:00
Code Issues Releases Wiki Activity

all in one

Browse Source
This commit is contained in:
Newnius 2019-12-26 16:50:25 +08:00
parent 47605ea654
commit ed9ebb3657
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@ -32,7 +32,7 @@ def launch_tasks(stats):
entries_to_remove = [] entries_to_remove = []
lock.acquire() lock.acquire()
for task_id, task in pending_tasks.items(): for task_id, task in pending_tasks.items():
if stats[task['gpus'][0]]['utilization_gpu'] < 75: if int(stats[task['gpus'][0]]['utilization_gpu']) < 75:
entries_to_remove.append(task_id) entries_to_remove.append(task_id)
script = " ".join([ script = " ".join([
"docker exec", "docker exec",