From 2e301ace17564980a5dc688f5119b057968be657 Mon Sep 17 00:00:00 2001 From: anjohan Date: Mon, 21 Oct 2024 13:23:09 -0600 Subject: [PATCH] auto extract pytorch version for libtorch download --- .github/workflows/tests.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index b717891..a0f1fed 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -26,7 +26,7 @@ jobs: python -m pip install --upgrade pip pip install torch --index-url https://download.pytorch.org/whl/cpu pip install pytest - wget -q https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-latest%2Bcpu.zip -O libtorch.zip + wget -q https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-$(pip show torch | awk '/Version/ {print $2;}')%2Bcpu.zip -O libtorch.zip unzip libtorch.zip - name: Install MPI run: |