From 993bd67f963651b5b02c42c676a343cc99577160 Mon Sep 17 00:00:00 2001 From: Jirka Borovec <6035284+Borda@users.noreply.github.com> Date: Thu, 1 Dec 2022 23:14:18 +0100 Subject: [PATCH] CI: prune dependency for benchmarks (#15879) * prune dependency for benchmarks * drop --- .azure/gpu-benchmark.yml | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/.azure/gpu-benchmark.yml b/.azure/gpu-benchmark.yml index bcd925e1d2..26b4a544e0 100644 --- a/.azure/gpu-benchmark.yml +++ b/.azure/gpu-benchmark.yml @@ -68,14 +68,7 @@ jobs: pip list displayName: 'Image info & NVIDIA' - - bash: | - python .actions/assistant.py requirements_prune_pkgs --packages [horovod,bagua,colossalai] --req_files [requirements/pytorch/strategies.txt] - - PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])") - python ./requirements/pytorch/adjust-versions.py requirements/pytorch/base.txt ${PYTORCH_VERSION} - displayName: 'Adjust dependencies' - - - bash: pip install -e .[dev,strategies,examples] --find-links ${TORCH_URL} + - bash: pip install -e .[dev] --find-links ${TORCH_URL} env: PACKAGE_NAME: "pytorch" FREEZE_REQUIREMENTS: "1"