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