Update pytorch version
This commit is contained in:
parent
c965ba9d3b
commit
6cfa588c0f
|
|
@ -31,7 +31,7 @@ jobs:
|
||||||
run: |
|
run: |
|
||||||
python -m pip install --upgrade pip
|
python -m pip install --upgrade pip
|
||||||
# cpu version of pytorch
|
# 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
|
# Install Atari Roms
|
||||||
pip install autorom
|
pip install autorom
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue