YAO/YAO-agent
1
0
Fork 0
mirror of https://github.com/newnius/YAO-agent.git synced 2025-12-12 21:16:44 +00:00
Code Issues Releases Wiki Activity

update

Browse Source
This commit is contained in:
Newnius
2020-05-26 23:00:02 +08:00
parent 5804c0ce72
commit f43af24ecd
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
agent.py
View File

@@ -211,6 +211,7 @@ class MyHandler(BaseHTTPRequestHandler):
exit_code, output = container.exec_run(['sh', '-c', script]) exit_code, output = container.exec_run(['sh', '-c', script])
msg = {"code": 0, "id": output.decode('utf-8').rstrip('\n')} msg = {"code": 0, "id": output.decode('utf-8').rstrip('\n')}
if docker_wait == "1":
lock.acquire() lock.acquire()
pending_tasks[token] = {'gpus': str(docker_gpus).split(','), 'gpu_mem': int(docker_gpu_mem)} pending_tasks[token] = {'gpus': str(docker_gpus).split(','), 'gpu_mem': int(docker_gpu_mem)}
id2token[msg['id']] = token id2token[msg['id']] = token