27 lines
1.0 KiB
Plaintext
27 lines
1.0 KiB
Plaintext
# 此Dockerfile适用于“无本地模型”的环境构建，如果需要使用chatglm等本地模型，请参考 docs/Dockerfile+ChatGLM
|
||
# 如何构建: 先修改 `config.py`， 然后 docker build -t gpt-academic-nolocal-vs -f docs/GithubAction+NoLocal+Vectordb .
|
||
# 如何运行: docker run --rm -it --net=host gpt-academic-nolocal-vs
|
||
FROM python:3.11
|
||
|
||
# 指定路径
|
||
WORKDIR /gpt
|
||
|
||
# 装载项目文件
|
||
COPY . .
|
||
|
||
# 安装依赖
|
||
RUN pip3 install -r requirements.txt
|
||
|
||
# 安装知识库插件的额外依赖
|
||
RUN apt-get update && apt-get install libgl1 -y
|
||
RUN pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cpu
|
||
RUN pip3 install transformers protobuf langchain sentence-transformers faiss-cpu nltk beautifulsoup4 bitsandbytes tabulate icetk --upgrade
|
||
RUN pip3 install unstructured[all-docs] --upgrade
|
||
|
||
# 可选步骤，用于预热模块
|
||
RUN python3 -c 'from check_proxy import warm_up_modules; warm_up_modules()'
|
||
RUN python3 -c 'from check_proxy import warm_up_vectordb; warm_up_vectordb()'
|
||
|
||
# 启动
|
||
CMD ["python3", "-u", "main.py"]
|