From 01ba14814f5a971ede028d96d8b20710fcd1a492 Mon Sep 17 00:00:00 2001 From: soon Date: Fri, 21 Apr 2023 08:57:10 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0Dockerfile?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Dockerfile | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 Dockerfile diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..79c029e --- /dev/null +++ b/Dockerfile @@ -0,0 +1,34 @@ +FROM python:3.8 + +MAINTAINER "chatGLM" + +COPY agent /chatGLM/agent + +COPY chains /chatGLM/chains + +COPY configs /chatGLM/configs + +COPY content /chatGLM/content + +COPY models /chatGLM/models + +COPY nltk_data /chatGLM/content + +COPY requirements.txt /chatGLM/ + +COPY cli_demo.py /chatGLM/ + +COPY webui.py /chatGLM/ + +WORKDIR /chatGLM + +RUN pip install --user torch torchvision tensorboard cython -i https://pypi.tuna.tsinghua.edu.cn/simple +# RUN pip install --user 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI' + +# RUN pip install --user 'git+https://github.com/facebookresearch/fvcore' +# install detectron2 +# RUN git clone https://github.com/facebookresearch/detectron2 + +RUN pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/ --trusted-host pypi.tuna.tsinghua.edu.cn + +CMD ["python","-u", "webui.py"]