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

add support for gpu specify

Browse Source
This commit is contained in:
Newnius 2019-04-12 16:54:33 +08:00
parent 5c90e5a94f
commit 21b1d2edd7
3 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Dockerfile
View File

@ -1,10 +1,7 @@
FROM nvidia/cuda:9.0-base FROM quickdeploy/yao-python3
MAINTAINER Newnius <newnius.cn@gmail.com> MAINTAINER Newnius <newnius.cn@gmail.com>
RUN apt update && \
apt install -y python3 python3-pip
RUN pip3 install docker kafka RUN pip3 install docker kafka
ADD bootstrap.sh /etc/bootstrap.sh ADD bootstrap.sh /etc/bootstrap.sh

5
bootstrap.sh
View File

@ -1,6 +1,9 @@
#!/usr/bin/env bash #!/usr/bin/env bash
# TODO: monitor the processes
python3 /root/yao-agent.py & python3 /root/yao-agent.py &
python3 /root/server.py python3 /root/server.py &
sleep infinity

6
server.py
View File

@ -75,14 +75,18 @@ class MyHandler(BaseHTTPRequestHandler):
'CONTENT_TYPE': self.headers['Content-Type'], 'CONTENT_TYPE': self.headers['Content-Type'],
}) })
docker_image = form["image"].value docker_image = form["image"].value
docker_name = form["name"].value
docker_cmd = form["cmd"].value docker_cmd = form["cmd"].value
docker_workspace = form["workspace"].value
docker_gpus = form["gpus"].value
try: try:
client = docker.from_env() client = docker.from_env()
container = client.containers.run( container = client.containers.run(
image=docker_image, image=docker_image,
hostname=docker_name,
command=docker_cmd, command=docker_cmd,
environment={"key": "value"}, environment={"repo": docker_workspace, "NVIDIA_VISIBLE_DEVICES": docker_gpus},
runtime="nvidia", runtime="nvidia",
detach=True detach=True
) )