add Dockerfile

pull/275/head
regmiao 2023-03-28 17:52:27 +08:00
parent 8798563db9
commit 89fc500237
1 changed files with 6 additions and 0 deletions

6
Dockerfile Normal file
View File

@ -0,0 +1,6 @@
FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
COPY . .
RUN pip3 install -r requirements.txt
RUN python3 pull_model.py
EXPOSE 7860
CMD [ "python","web_demo.py" ]