Update pytorch version

Stable-Baselines-Team · Nov 16, 2023 · a180dee · a180dee
1 parent 3ad67ee
commit a180dee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -31,7 +31,7 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         # cpu version of pytorch
-        pip install torch==1.13+cpu -f https://download.pytorch.org/whl/torch_stable.html
+        pip install torch==2.1.1 --index-url https://download.pytorch.org/whl/cpu
 
         # Install Atari Roms
         pip install autorom