From e31a67541b3a323a4772f92bd54298005a607cee Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Mon, 31 Oct 2022 14:10:08 +0100 Subject: [PATCH] ci: fix install lite on GPUs (#15375) --- .azure/gpu-tests-lite.yml | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/.azure/gpu-tests-lite.yml b/.azure/gpu-tests-lite.yml index ae88f9711d..ceb4c671cf 100644 --- a/.azure/gpu-tests-lite.yml +++ b/.azure/gpu-tests-lite.yml @@ -63,17 +63,13 @@ jobs: - bash: | set -e - PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])") - python ./requirements/pytorch/adjust-versions.py requirements/lite/base.txt ${PYTORCH_VERSION} - CUDA_VERSION_MM=$(python -c "import torch ; print(''.join(map(str, torch.version.cuda.split('.')[:2])))") pip install -e .[dev,strategies] --find-links https://download.pytorch.org/whl/cu${CUDA_VERSION_MM}/torch_stable.html - pip list env: - PACKAGE_NAME: pytorch - FREEZE_REQUIREMENTS: 1 - displayName: 'Install dependencies' + PACKAGE_NAME: "lite" + FREEZE_REQUIREMENTS: "1" + displayName: 'Install package & dependencies' - bash: | set -e