Update pytorch version

This commit is contained in:
Antonin Raffin 2023-11-16 18:00:31 +01:00 committed by Andreas Schaarschmidt
parent c965ba9d3b
commit 6cfa588c0f
1 changed files with 1 additions and 1 deletions

View File

@ -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