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

all in one

Browse Source
This commit is contained in:
Newnius 2019-12-26 18:47:25 +08:00
parent a0ff4274d0
commit 702fc8dcfb
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@ -131,7 +131,7 @@ class MyHandler(BaseHTTPRequestHandler):
try:
script = " ".join([
"docker run",
"--gpus '\'\"device=" + docker_gpus + "\"$'\''",
"--gpus '\"device=" + docker_gpus + "\"'",
"--detach=True",
"--hostname " + docker_name,
"--network " + docker_network,
@ -143,11 +143,9 @@ class MyHandler(BaseHTTPRequestHandler):
docker_cmd
])
print('sh -c \'' + script + '\'')
client = docker.from_env()
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')}
print(msg)