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 19:24:18 +08:00
parent 6cf49f9285
commit 10f69f2ea7
3 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -18,6 +18,8 @@ ADD monitor.py /root/monitor.py
ADD executor.py /root/executor.py ADD executor.py /root/executor.py
ADD main.py /root/main.py
WORKDIR /root WORKDIR /root
CMD ["/etc/bootstrap.sh"] CMD ["/etc/bootstrap.sh"]

8
bootstrap.sh
View File

@ -5,8 +5,10 @@
# run nvidia-smi in background to speed up the query and reduce CPU load (why?) # run nvidia-smi in background to speed up the query and reduce CPU load (why?)
nvidia-smi daemon nvidia-smi daemon
python3 /root/monitor.py & #python3 /root/monitor.py &
python3 /root/executor.py & #python3 /root/executor.py &
sleep infinity #sleep infinity
python3 /root/main.py

4
main.py
View File

@ -143,18 +143,14 @@ class MyHandler(BaseHTTPRequestHandler):
docker_cmd docker_cmd
]) ])
print(script)
client = docker.from_env() client = docker.from_env()
container = client.containers.get('yao-agent-helper') container = client.containers.get('yao-agent-helper')
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')}
print(msg)
lock.acquire() lock.acquire()
pending_tasks[msg['id']] = {'gpus': str(docker_gpus).split(',')} pending_tasks[msg['id']] = {'gpus': str(docker_gpus).split(',')}
lock.release() lock.release()
print(msg)
if exit_code != 0: if exit_code != 0:
msg["code"] = 1 msg["code"] = 1
except Exception as e: except Exception as e: