From 7f4f59815213455871a262453cf490534ed81fb8 Mon Sep 17 00:00:00 2001 From: Maarten de Vries Date: Sat, 30 Mar 2024 12:13:15 +0100 Subject: [PATCH] Bump torch version installed in CI. --- .github/workflows/rust.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/rust.yml b/.github/workflows/rust.yml index 2b0b027..0bd9c4f 100644 --- a/.github/workflows/rust.yml +++ b/.github/workflows/rust.yml @@ -10,7 +10,7 @@ jobs: steps: - name: Install pytorch run: | - pip3 install torch==2.0.0 torchvision==0.15.0 --index-url https://download.pytorch.org/whl/cpu + pip3 install torch==2.2.0 --index-url https://download.pytorch.org/whl/cpu echo LIBTORCH_USE_PYTORCH=1 >> "$GITHUB_ENV" echo LD_LIBRARY_PATH="$(python -m site --user-site)/torch/lib${LD_LIBRARY_PATH:+:}$LD_LIBRARY_PATH" >> "$GITHUB_ENV" - name: Checkout code