From 6cfa588c0fa25876a9457458472291c388b0f177 Mon Sep 17 00:00:00 2001 From: Antonin Raffin Date: Thu, 16 Nov 2023 18:00:31 +0100 Subject: [PATCH] Update pytorch version --- .github/workflows/ci.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 048fb2a..6f81953 100644 --- 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