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

update

Browse Source
This commit is contained in:
Newnius 2020-04-10 17:55:49 +08:00
parent 8b90351f6a
commit a309e0f509
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
agent.py
View File

@ -38,7 +38,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 int(utils[task['gpus'][0]]) < 70: if int(utils[task['gpus'][0]]) < 60:
entries_to_remove.append(task_id) entries_to_remove.append(task_id)
script = " ".join([ script = " ".join([
"docker exec", "docker exec",
@ -97,7 +97,7 @@ class MyHandler(BaseHTTPRequestHandler):
'state': container.attrs['State'] 'state': container.attrs['State']
} }
if container_id in pending_tasks: if container_id in pending_tasks:
status['status'] = 'Ready' status['status'] = 'ready'
if status['command'] is not None: if status['command'] is not None:
status['command'] = ' '.join(container.attrs['Config']['Cmd']) status['command'] = ' '.join(container.attrs['Config']['Cmd'])
msg = {'code': 0, 'status': status} msg = {'code': 0, 'status': status}
@ -311,7 +311,7 @@ def report_msg(stats):
flag = True flag = True
if abs(last_version['status'][i]['memory_free'] - post_fields['status'][i]['memory_free']) / post_fields['status'][i]['memory_total'] > 0.05: if abs(last_version['status'][i]['memory_free'] - post_fields['status'][i]['memory_free']) / post_fields['status'][i]['memory_total'] > 0.05:
flag = True flag = True
if abs(last_version['status'][i]['utilization_gpu'] - post_fields['status'][i]['utilization_gpu']) > 15.0: if abs(last_version['status'][i]['utilization_gpu'] - post_fields['status'][i]['utilization_gpu']) > 10.0:
flag = True flag = True
else: else:
flag = True flag = True