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:13:13 +08:00
parent 03aa692227
commit 69fd081e29
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@ -131,7 +131,7 @@ class MyHandler(BaseHTTPRequestHandler):
try: try:
script = " ".join([ script = " ".join([
"docker run", "docker run",
"--gpus '\"device=" + docker_gpus + "\"'", "--gpus '\\\"device=" + docker_gpus + "\\\"'",
"--detach=True", "--detach=True",
"--hostname " + docker_name, "--hostname " + docker_name,
"--network " + docker_network, "--network " + docker_network,
@ -147,7 +147,7 @@ class MyHandler(BaseHTTPRequestHandler):
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) print(msg)