1
0
mirror of https://github.com/newnius/YAO-Dockerfiles.git synced 2025-06-06 07:51:54 +00:00
YAO-Dockerfiles/yao-optimizer/dev/Dockerfile
2020-05-02 20:24:39 +08:00

21 lines
496 B
Docker

FROM tensorflow/tensorflow:1.14.0-gpu
MAINTAINER Newnius <newnius.cn@gmail.com>
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 keras
ADD bootstrap.sh /etc/bootstrap.sh
RUN git clone https://code.newnius.com/newnius/yao-optimizer.git /root/yao-optimizer/
WORKDIR /root/yao-optimizer
CMD ["/etc/bootstrap.sh"]