mirror of
https://github.com/csunny/DB-GPT.git
synced 2026-01-16 07:26:31 +00:00
b5f2758dc0
fix: Fix error CICD config fix: Fix docker image error fix: Reduce docker image size fix: remove other architecture fix: remove llama.cpp dependencies from base image
60 lines
2.1 KiB
Docker
60 lines
2.1 KiB
Docker
ARG BASE_IMAGE="nvidia/cuda:11.8.0-runtime-ubuntu22.04"
|
||
|
||
FROM ${BASE_IMAGE}
|
||
ARG BASE_IMAGE
|
||
|
||
RUN apt-get update && apt-get install -y git python3 pip wget sqlite3 \
|
||
&& apt-get clean
|
||
|
||
ARG BUILD_LOCAL_CODE="false"
|
||
ARG LANGUAGE="en"
|
||
ARG PIP_INDEX_URL="https://pypi.org/simple"
|
||
ENV PIP_INDEX_URL=$PIP_INDEX_URL
|
||
|
||
RUN mkdir -p /app
|
||
|
||
# COPY only requirements.txt first to leverage Docker cache
|
||
COPY ./requirements.txt /app/requirements.txt
|
||
COPY ./setup.py /app/setup.py
|
||
COPY ./README.md /app/README.md
|
||
|
||
WORKDIR /app
|
||
|
||
# ENV CMAKE_ARGS="-DLLAMA_CUBLAS=ON -DLLAMA_AVX2=OFF -DLLAMA_F16C=OFF -DLLAMA_FMA=OFF"
|
||
# ENV FORCE_CMAKE=1
|
||
# Install all
|
||
# RUN pip3 install -i $PIP_INDEX_URL ".[all]"
|
||
|
||
RUN pip3 install --upgrade pip -i $PIP_INDEX_URL \
|
||
&& pip3 install -i $PIP_INDEX_URL . \
|
||
# && pip3 install -i $PIP_INDEX_URL ".[llama_cpp]" \
|
||
&& (if [ "${LANGUAGE}" = "zh" ]; \
|
||
# language is zh, download zh_core_web_sm from github
|
||
then wget https://github.com/explosion/spacy-models/releases/download/zh_core_web_sm-3.5.0/zh_core_web_sm-3.5.0-py3-none-any.whl -O /tmp/zh_core_web_sm-3.5.0-py3-none-any.whl \
|
||
&& pip3 install /tmp/zh_core_web_sm-3.5.0-py3-none-any.whl -i $PIP_INDEX_URL \
|
||
&& rm /tmp/zh_core_web_sm-3.5.0-py3-none-any.whl; \
|
||
# not zh, download directly
|
||
else python3 -m spacy download zh_core_web_sm; \
|
||
fi;) \
|
||
&& rm -rf `pip3 cache dir`
|
||
|
||
ARG BUILD_LOCAL_CODE="false"
|
||
# COPY the rest of the app
|
||
COPY . /app
|
||
|
||
# TODO：Need to find a better way to determine whether to build docker image with local code.
|
||
RUN (if [ "${BUILD_LOCAL_CODE}" = "true" ]; \
|
||
then rm -rf /app/logs && rm -rf /app/pilot/data && rm -rf /app/pilot/message; \
|
||
fi;)
|
||
|
||
ARG LOAD_EXAMPLES="true"
|
||
|
||
RUN (if [ "${LOAD_EXAMPLES}" = "true" ]; \
|
||
then mkdir -p /app/pilot/data && sqlite3 /app/pilot/data/default_sqlite.db < /app/docker/examples/sqls/case_1_student_manager_sqlite.sql \
|
||
&& sqlite3 /app/pilot/data/default_sqlite.db < /app/docker/examples/sqls/case_2_ecom_sqlite.sql \
|
||
&& sqlite3 /app/pilot/data/default_sqlite.db < /app/docker/examples/sqls/test_case_info_sqlite.sql; \
|
||
fi;)
|
||
|
||
EXPOSE 5000
|
||
|
||
CMD ["python3", "pilot/server/dbgpt_server.py"] |