Update pytorch version
This commit is contained in:
parent
c965ba9d3b
commit
6cfa588c0f
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue