diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..88d1875 --- /dev/null +++ b/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 pandas sklearn tensorflow-gpu==1.14 + +ADD bootstrap.sh /etc/bootstrap.sh + +ADD agent.py /root/agent.py + +WORKDIR /root + +CMD ["/etc/bootstrap.sh"] \ No newline at end of file