YAO/YAO-agent
1
0
Fork 0
mirror of https://github.com/newnius/YAO-agent.git synced 2025-12-15 05:56:43 +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
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>
RUN apt update && \
apt install -y python3 python3-pip
RUN pip3 install docker kafka
ADD bootstrap.sh /etc/bootstrap.sh