mirror of
https://github.com/newnius/YAO-optimizer.git
synced 2025-06-06 22:51:55 +00:00
add files
This commit is contained in:
parent
dca26ecd3d
commit
a19c11fc68
21
Dockerfile
Normal file
21
Dockerfile
Normal file
@ -0,0 +1,21 @@
|
||||
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
|
||||
|
||||
ADD agent.py /root/agent.py
|
||||
|
||||
WORKDIR /root
|
||||
|
||||
CMD ["/etc/bootstrap.sh"]
|
Loading…
Reference in New Issue
Block a user