From 0d6783674db82dc58d2211d36c769e955bb157ca Mon Sep 17 00:00:00 2001 From: vfdev Date: Fri, 27 Sep 2024 10:39:13 +0200 Subject: [PATCH] Fixed pytorch cpu installation link to avoid downloading nvidia libs --- .github/workflows/hvd-tests.yml | 2 +- .github/workflows/install_docs_deps.sh | 2 +- .github/workflows/unit-tests.yml | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/hvd-tests.yml b/.github/workflows/hvd-tests.yml index 35e107f888b..2733250faf8 100644 --- a/.github/workflows/hvd-tests.yml +++ b/.github/workflows/hvd-tests.yml @@ -62,7 +62,7 @@ jobs: run: | #install other dependencies - pip install torch torchvision -f https://download.pytorch.org/whl/cpu/torch_stable.html + pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu pip install -r requirements-dev.txt pip install horovod python setup.py install diff --git a/.github/workflows/install_docs_deps.sh b/.github/workflows/install_docs_deps.sh index 52ba063cef2..499b6421194 100644 --- a/.github/workflows/install_docs_deps.sh +++ b/.github/workflows/install_docs_deps.sh @@ -1,6 +1,6 @@ # remove pkg-resources as it causes failure when installing https://github.com/pytorch-ignite/sphinxcontrib-versioning pip uninstall -y pkg-resources setuptools && pip install --upgrade setuptools pip wheel -pip install torch torchvision -f https://download.pytorch.org/whl/cpu/torch_stable.html -U +pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu -U pip install -r requirements-dev.txt pip install -r docs/requirements.txt pip install git+https://github.com/pytorch-ignite/sphinxcontrib-versioning.git diff --git a/.github/workflows/unit-tests.yml b/.github/workflows/unit-tests.yml index 0b94e0d0e9e..feacc642ffb 100644 --- a/.github/workflows/unit-tests.yml +++ b/.github/workflows/unit-tests.yml @@ -83,11 +83,11 @@ jobs: - name: Install PyTorch if: ${{ matrix.pytorch-channel == 'pytorch' }} - run: pip install torch torchvision -f https://download.pytorch.org/whl/cpu/torch_stable.html + run: pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu - name: Install PyTorch (nightly) if: ${{ matrix.pytorch-channel == 'pytorch-nightly' }} - run: pip install torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html --pre + run: pip install torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu --pre - name: Install dependencies run: |