add support for gpu specify

Dockerfile

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

bootstrap.sh

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

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
 				)