From 123a999bbe5a45f4bdaec7d5add2c6fa6abe4af8 Mon Sep 17 00:00:00 2001 From: Newnius Date: Sat, 23 May 2020 18:29:02 +0800 Subject: [PATCH] add yao-agent --- yao-agent/dev/Dockerfile | 21 +++++++++++++++++++++ yao-agent/dev/bootstrap.sh | 6 ++++++ yao-agent/mock/Dockerfile | 21 +++++++++++++++++++++ yao-agent/mock/bootstrap.sh | 6 ++++++ 4 files changed, 54 insertions(+) create mode 100644 yao-agent/dev/Dockerfile create mode 100755 yao-agent/dev/bootstrap.sh create mode 100644 yao-agent/mock/Dockerfile create mode 100755 yao-agent/mock/bootstrap.sh diff --git a/yao-agent/dev/Dockerfile b/yao-agent/dev/Dockerfile new file mode 100644 index 0000000..3697479 --- /dev/null +++ b/yao-agent/dev/Dockerfile @@ -0,0 +1,21 @@ +FROM tensorflow/tensorflow:1.14.0-gpu + +MAINTAINER Newnius + +RUN apt update && \ + apt install -y python3 python3-pip && \ + rm -rf /var/lib/apt/lists/* + +RUN apt update && \ + apt install -y git vim httpie && \ + rm -rf /var/lib/apt/lists/* + +RUN pip3 install docker kafka psutil + +RUN git clone https://github.com/newnius/YAO-agennt.git /yao-agent + +WORKDIR /yao-agent + +ADD bootstrap.sh /etc/ + +CMD ["/etc/bootstrap.sh"] diff --git a/yao-agent/dev/bootstrap.sh b/yao-agent/dev/bootstrap.sh new file mode 100755 index 0000000..3769ad1 --- /dev/null +++ b/yao-agent/dev/bootstrap.sh @@ -0,0 +1,6 @@ +#!/usr/bin/env bash + +# run nvidia-smi in background to speed up the query and reduce CPU load (why?) +nvidia-smi daemon + +python3 /yao-agent/agent.py diff --git a/yao-agent/mock/Dockerfile b/yao-agent/mock/Dockerfile new file mode 100644 index 0000000..3697479 --- /dev/null +++ b/yao-agent/mock/Dockerfile @@ -0,0 +1,21 @@ +FROM tensorflow/tensorflow:1.14.0-gpu + +MAINTAINER Newnius + +RUN apt update && \ + apt install -y python3 python3-pip && \ + rm -rf /var/lib/apt/lists/* + +RUN apt update && \ + apt install -y git vim httpie && \ + rm -rf /var/lib/apt/lists/* + +RUN pip3 install docker kafka psutil + +RUN git clone https://github.com/newnius/YAO-agennt.git /yao-agent + +WORKDIR /yao-agent + +ADD bootstrap.sh /etc/ + +CMD ["/etc/bootstrap.sh"] diff --git a/yao-agent/mock/bootstrap.sh b/yao-agent/mock/bootstrap.sh new file mode 100755 index 0000000..5d384c8 --- /dev/null +++ b/yao-agent/mock/bootstrap.sh @@ -0,0 +1,6 @@ +#!/usr/bin/env bash + +# run nvidia-smi in background to speed up the query and reduce CPU load (why?) +nvidia-smi daemon + +python3 /yao-agent/mock.py