Use +cpu version

Stable-Baselines-Team · Oct 24, 2024 · 94af853 · 94af853
1 parent 6e37805
commit 94af853
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -34,7 +34,7 @@ jobs:
           pip install uv
           # cpu version of pytorch
           # See https://github.com/astral-sh/uv/issues/1497
-          uv pip install --system torch==2.5.0.* --index https://download.pytorch.org/whl/cpu
+          uv pip install --system torch==2.5.0+cpu --index https://download.pytorch.org/whl/cpu
 
           # Install Atari Roms
           uv pip install --system autorom