From 3b4f8488e815bd9cbc2efe56547bc9845cbe72ee Mon Sep 17 00:00:00 2001
From: hhhhsc701 <56435672+hhhhsc701@users.noreply.github.com>
Date: Wed, 17 Dec 2025 11:49:47 +0800
Subject: [PATCH] fix: update Dockerfile to improve pip installation process
 and remove unnecessary uninstalls (#173)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* feature: unstructured支持简单pdf处理

* feature: update values.yaml to enhance ray-cluster configuration with security context, environment variables, and resource limits

* fix: update Dockerfile to improve pip installation process and remove unnecessary uninstalls
---
 scripts/images/runtime/Dockerfile | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/scripts/images/runtime/Dockerfile b/scripts/images/runtime/Dockerfile
index 1794d0c..d1f886a 100644
--- a/scripts/images/runtime/Dockerfile
+++ b/scripts/images/runtime/Dockerfile
@@ -20,10 +20,9 @@ ENV PYTHONPATH=/opt/runtime/datamate/
 WORKDIR /opt/runtime
 
 RUN --mount=type=cache,target=/root/.cache/uv \
-    uv pip install -e . --system \
+    UV_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cpu" uv pip install -e . --system --index-strategy unsafe-best-match \
     && UV_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cpu" uv pip install -r /opt/runtime/datamate/ops/pyproject.toml --system \
-    && uv pip uninstall torch torchvision triton --system \
-    && uv pip list | grep -E '^nvidia-' | awk '{print $1}' | xargs -r uv pip uninstall --system \
+    && uv pip uninstall torch torchvision --system \
     && python -m spacy download zh_core_web_sm
 
 RUN ln -sf /usr/share/zoneinfo/Asia/Shanghai /etc/localtime \