From eb5579dbbdec160370dc9f0f7be349b991f7c6de Mon Sep 17 00:00:00 2001
From: "2569718930@qq.com" <2569718930@qq.com>
Date: Tue, 16 Jun 2026 02:04:52 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=20torch=20=E5=AE=89=E8=A3=85?=
 =?UTF-8?q?=EF=BC=9A=E4=BD=BF=E7=94=A8=20extra-index-url=20=E6=9B=BF?=
 =?UTF-8?q?=E4=BB=A3=E7=9B=B4=E6=8E=A5=20URL?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .github/workflows/ci.yml | 2 +-
 Dockerfile               | 2 +-
 requirements.lock        | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 8a8e1b1f..403d61a7 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -21,7 +21,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install -r requirements.lock -r requirements-dev.lock
+          pip install --extra-index-url https://download.pytorch.org/whl/cpu -r requirements.lock -r requirements-dev.lock
 
       - name: Ruff
         run: python -m ruff check .
diff --git a/Dockerfile b/Dockerfile
index 2d05e79b..48335438 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -18,7 +18,7 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
 COPY requirements.lock .
 
 RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install --prefer-binary -r requirements.lock
+    pip install --prefer-binary --extra-index-url https://download.pytorch.org/whl/cpu -r requirements.lock
 
 COPY . .
 
diff --git a/requirements.lock b/requirements.lock
index 571ec7f2..a6f3cd1d 100644
--- a/requirements.lock
+++ b/requirements.lock
@@ -110,7 +110,7 @@ netcdf4==1.7.4
     # via -r requirements.txt
 numpy==2.4.6
 
-torch @ https://download.pytorch.org/whl/cpu/torch-2.5.1%2Bcpu-cp311-cp311-manylinux_2_28_x86_64.whl ; python_version >= "3.11"
+torch
 typing_extensions==4.15.0
     # via torch
     # via