Fix for pytorch install

Stable-Baselines-Team · Oct 24, 2024 · 6e37805 · 6e37805
1 parent 497f5fe
commit 6e37805
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -33,7 +33,8 @@ jobs:
           # Use uv for faster downloads
           pip install uv
           # cpu version of pytorch
-          uv pip install --system torch==2.1.1 --index-url https://download.pytorch.org/whl/cpu
+          # See https://github.com/astral-sh/uv/issues/1497
+          uv pip install --system torch==2.5.0.* --index https://download.pytorch.org/whl/cpu
 
           # Install Atari Roms
           uv pip install --system autorom