From d3e59bc7f30aee0faf57742c04f19ea1faff8c84 Mon Sep 17 00:00:00 2001
From: csf123321 <sifanchen2004@outlook.com>
Date: Thu, 2 Apr 2026 13:58:59 +0800
Subject: [PATCH] =?UTF-8?q?=E5=BC=BA=E5=88=B6cpu=20rtorch?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/pyproject.toml | 15 ++++++++++++---
 dockerfile             |  2 +-
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/backend/pyproject.toml b/backend/pyproject.toml
index d53646a..b52f0be 100644
--- a/backend/pyproject.toml
+++ b/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"
diff --git a/dockerfile b/dockerfile
index 9b99ecf..02e96aa 100644
--- a/dockerfile
+++ b/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