From 21b1d2edd7db7e092a02aed2cfe81c61531b2e78 Mon Sep 17 00:00:00 2001 From: Newnius Date: Fri, 12 Apr 2019 16:54:33 +0800 Subject: [PATCH] add support for gpu specify --- Dockerfile | 5 +---- bootstrap.sh | 5 ++++- server.py | 6 +++++- 3 files changed, 10 insertions(+), 6 deletions(-) diff --git a/Dockerfile b/Dockerfile index 86fd624..1206461 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,10 +1,7 @@ -FROM nvidia/cuda:9.0-base +FROM quickdeploy/yao-python3 MAINTAINER Newnius -RUN apt update && \ - apt install -y python3 python3-pip - RUN pip3 install docker kafka ADD bootstrap.sh /etc/bootstrap.sh diff --git a/bootstrap.sh b/bootstrap.sh index 7a61e58..94f8c2b 100755 --- a/bootstrap.sh +++ b/bootstrap.sh @@ -1,6 +1,9 @@ #!/usr/bin/env bash +# TODO: monitor the processes python3 /root/yao-agent.py & -python3 /root/server.py \ No newline at end of file +python3 /root/server.py & + +sleep infinity \ No newline at end of file diff --git a/server.py b/server.py index 132f18d..3ee7873 100644 --- a/server.py +++ b/server.py @@ -75,14 +75,18 @@ class MyHandler(BaseHTTPRequestHandler): 'CONTENT_TYPE': self.headers['Content-Type'], }) docker_image = form["image"].value + docker_name = form["name"].value docker_cmd = form["cmd"].value + docker_workspace = form["workspace"].value + docker_gpus = form["gpus"].value try: client = docker.from_env() container = client.containers.run( image=docker_image, + hostname=docker_name, command=docker_cmd, - environment={"key": "value"}, + environment={"repo": docker_workspace, "NVIDIA_VISIBLE_DEVICES": docker_gpus}, runtime="nvidia", detach=True )