Files
2018-06-07 12:47:18 -07:00

10 lines
156 B
Docker

ARG VERSION
FROM tensorflow/tensorflow:$VERSION
WORKDIR /
RUN pip install keras h5py
COPY /trainer /trainer
ENTRYPOINT ["python", "-m", "trainer.start"]