强制cpu rtorch

backend/pyproject.toml

@@ -49,7 +49,6 @@ dependencies = [
     "safetensors==0.7.0",
     "scikit-learn==1.8.0",
     "scipy==1.17.1",
-    "sentence-transformers==5.2.3",
     "shellingham==1.5.4",
     "sniffio==1.3.1",
     "sqlalchemy==2.0.48",
@@ -57,8 +56,6 @@ dependencies = [
     "sympy==1.14.0",
     "threadpoolctl==3.6.0",
     "tokenizers==0.22.2",
-    "torch==2.10.0",
-    "torchvision==0.25.0",
     "tqdm==4.67.3",
     "transformers==5.3.0",
     "typer==0.24.1",
@@ -68,4 +65,16 @@ dependencies = [
     "tzlocal==5.3.1",
     "urllib3==2.6.3",
     "uvicorn==0.41.0",
+    "torch==2.11.0+cpu",
+    "torchvision==0.26.0+cpu",
+    "torchaudio==2.11.0+cpu",
+    "sentence-transformers>=5.3.0",
 ]
+
+[[tool.uv.index]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+default = false
+
+[tool.uv]
+index-strategy = "unsafe-best-match"

dockerfile

@@ -20,7 +20,7 @@ WORKDIR /backend
 COPY backend/pyproject.toml backend/uv.lock ./
 RUN --mount=type=cache,target=/root/.cache/uv \
     pip install --no-cache-dir uv && \
-    uv sync --frozen --no-dev
+    uv sync --frozen --no-dev --index https://pypi.tuna.tsinghua.edu.cn/simple/
 
 # 复制后端代码
 COPY backend/app ./app