1
0
mirror of https://github.com/newnius/YAO-Dockerfiles.git synced 2025-06-07 08:11:56 +00:00
YAO-Dockerfiles/tensorflow/2.1-gpu/Dockerfile
2020-05-04 17:50:07 +08:00

18 lines
330 B
Docker

FROM tensorflow/tensorflow:2.1.0-gpu-py3
MAINTAINER Newnius <newnius.cn@gmail.com>
USER root
RUN apt update && \
apt install -y git && \
rm -rf /var/lib/apt/lists/*
RUN pip3 install requests psutil py-cpuinfo pandas typing hdfs
ADD bootstrap.sh /etc/bootstrap.sh
ADD save.py /etc/save.py
ENTRYPOINT ["/etc/bootstrap.sh"]