Add a Dockerfile and document usage in README

This commit is contained in:
Madison May
2019-02-14 18:22:14 -05:00
committed by Jeff Wu
parent 23ed990b03
commit 99af6d7092
3 changed files with 50 additions and 0 deletions

12
Dockerfile.cpu Normal file
View File

@ -0,0 +1,12 @@
FROM tensorflow/tensorflow:1.12.0-py3
ENV LANG=C.UTF-8
RUN mkdir /gpt-2
WORKDIR /gpt-2
COPY requirements.txt download_model.sh /gpt-2/
RUN apt-get update && \
apt-get install -y curl && \
sh download_model.sh 117M
RUN pip3 install -r requirements.txt
ADD . /gpt-2