diff --git a/.azure/gpu-tests-fabric.yml b/.azure/gpu-tests-fabric.yml index d5980be0c48d4..da0483dbdae53 100644 --- a/.azure/gpu-tests-fabric.yml +++ b/.azure/gpu-tests-fabric.yml @@ -83,9 +83,8 @@ jobs: displayName: "set env. vars" - bash: | echo "##vso[task.setvariable variable=TORCH_URL]https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM}" - echo "##vso[task.setvariable variable=TORCHVISION_URL]https://download.pytorch.org/whl/test/cu124/torchvision-0.19.0%2Bcu124-cp${PYTHON_VERSION_MM}-cp${PYTHON_VERSION_MM}-linux_x86_64.whl" condition: endsWith(variables['Agent.JobName'], 'future') - displayName: "set env. vars 4 future" + displayName: "extend env. vars 4 future" - bash: | echo $(DEVICES) @@ -113,7 +112,7 @@ jobs: - bash: | set -e extra=$(python -c "print({'lightning': 'fabric-'}.get('$(PACKAGE_NAME)', ''))") - pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}" --find-links="${TORCHVISION_URL}" + pip install -e ".[${extra}dev]" pytest-timeout -U --extra-index-url="${TORCH_URL}" pip install setuptools==75.6.0 jsonargparse==4.35.0 displayName: "Install package & dependencies" diff --git a/.azure/gpu-tests-pytorch.yml b/.azure/gpu-tests-pytorch.yml index b763636ca5740..81ed24bbe3806 100644 --- a/.azure/gpu-tests-pytorch.yml +++ b/.azure/gpu-tests-pytorch.yml @@ -87,9 +87,8 @@ jobs: displayName: "set env. vars" - bash: | echo "##vso[task.setvariable variable=TORCH_URL]https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM}" - echo "##vso[task.setvariable variable=TORCHVISION_URL]https://download.pytorch.org/whl/test/cu124/torchvision-0.19.0%2Bcu124-cp${PYTHON_VERSION_MM}-cp${PYTHON_VERSION_MM}-linux_x86_64.whl" condition: endsWith(variables['Agent.JobName'], 'future') - displayName: "set env. vars 4 future" + displayName: "extend env. vars 4 future" - bash: | echo $(DEVICES) @@ -117,7 +116,7 @@ jobs: - bash: | set -e extra=$(python -c "print({'lightning': 'pytorch-'}.get('$(PACKAGE_NAME)', ''))") - pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}" --find-links="${TORCHVISION_URL}" + pip install -e ".[${extra}dev]" pytest-timeout -U --extra-index-url="${TORCH_URL}" pip install setuptools==75.6.0 jsonargparse==4.35.0 displayName: "Install package & dependencies"