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:01:05 +08:00
parent 702fc8dcfb
commit caccd74462
1 changed files with 4 additions and 2 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,9 +143,11 @@ 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)