2020-04-30 00:32:41 +08:00
|
|
|
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
|
|
|
|
|
|
|
|
|
|
ADD bootstrap.sh /etc/bootstrap.sh
|
|
|
|
|
|
2020-05-01 16:43:41 +08:00
|
|
|
RUN mkdir /root/data/
|
|
|
|
|
ADD serve.py /root/serve.py
|
|
|
|
|
ADD model_tensorflow.py /root/model_tensorflow.py
|
2020-04-30 00:32:41 +08:00
|
|
|
|
|
|
|
|
WORKDIR /root
|
|
|
|
|
|
|
|
|
|
CMD ["/etc/bootstrap.sh"]
|
