diff --git a/Dockerfile b/Dockerfile index 7c518d411..4bd8ec94f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -59,6 +59,7 @@ RUN --mount=type=cache,id=ragflow_apt,target=/var/cache/apt,sharing=locked \ apt install -y default-jdk && \ apt install -y libatk-bridge2.0-0 && \ apt install -y libpython3-dev libgtk-4-1 libnss3 xdg-utils libgbm-dev && \ + apt install -y libjemalloc-dev && \ apt install -y python3-pip pipx nginx unzip curl wget git vim less RUN if [ "$NEED_MIRROR" == "1" ]; then \ diff --git a/docker/entrypoint-parser.sh b/docker/entrypoint-parser.sh index 899f16f8f..07dc241f8 100755 --- a/docker/entrypoint-parser.sh +++ b/docker/entrypoint-parser.sh @@ -15,8 +15,9 @@ CONSUMER_NO_BEG=$1 CONSUMER_NO_END=$2 function task_exe(){ + JEMALLOC_PATH=$(pkg-config --variable=libdir jemalloc)/libjemalloc.so while [ 1 -eq 1 ]; do - $PY rag/svr/task_executor.py $1; + LD_PRELOAD=$JEMALLOC_PATH $PY rag/svr/task_executor.py $1; done } diff --git a/docker/entrypoint.sh b/docker/entrypoint.sh index 388102960..37bfb29b6 100755 --- a/docker/entrypoint.sh +++ b/docker/entrypoint.sh @@ -17,8 +17,9 @@ if [[ -z "$WS" || $WS -lt 1 ]]; then fi function task_exe(){ + JEMALLOC_PATH=$(pkg-config --variable=libdir jemalloc)/libjemalloc.so while [ 1 -eq 1 ];do - $PY rag/svr/task_executor.py $1; + LD_PRELOAD=$JEMALLOC_PATH $PY rag/svr/task_executor.py $1; done }