From 89fc500237ec7292a3291c26d086707cc3b2b831 Mon Sep 17 00:00:00 2001 From: regmiao Date: Tue, 28 Mar 2023 17:52:27 +0800 Subject: [PATCH] add Dockerfile --- Dockerfile | 6 ++++++ 1 file changed, 6 insertions(+) create mode 100644 Dockerfile diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..bad6354 --- /dev/null +++ b/Dockerfile @@ -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" ] \ No newline at end of file