Skip to content

Commit 5e9009b

Browse files
committed
fix
1 parent 97cf6ed commit 5e9009b

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

.azure/gpu-tests-fabric.yml

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -83,9 +83,8 @@ jobs:
8383
displayName: "set env. vars"
8484
- bash: |
8585
echo "##vso[task.setvariable variable=TORCH_URL]https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM}"
86-
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"
8786
condition: endsWith(variables['Agent.JobName'], 'future')
88-
displayName: "set env. vars 4 future"
87+
displayName: "extend env. vars 4 future"
8988
9089
- bash: |
9190
echo $(DEVICES)
@@ -113,7 +112,7 @@ jobs:
113112
- bash: |
114113
set -e
115114
extra=$(python -c "print({'lightning': 'fabric-'}.get('$(PACKAGE_NAME)', ''))")
116-
pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}" --find-links="${TORCHVISION_URL}"
115+
pip install -e ".[${extra}dev]" pytest-timeout -U --extra-index-url="${TORCH_URL}"
117116
pip install setuptools==75.6.0 jsonargparse==4.35.0
118117
displayName: "Install package & dependencies"
119118

.azure/gpu-tests-pytorch.yml

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -87,9 +87,8 @@ jobs:
8787
displayName: "set env. vars"
8888
- bash: |
8989
echo "##vso[task.setvariable variable=TORCH_URL]https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM}"
90-
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"
9190
condition: endsWith(variables['Agent.JobName'], 'future')
92-
displayName: "set env. vars 4 future"
91+
displayName: "extend env. vars 4 future"
9392
9493
- bash: |
9594
echo $(DEVICES)
@@ -117,7 +116,7 @@ jobs:
117116
- bash: |
118117
set -e
119118
extra=$(python -c "print({'lightning': 'pytorch-'}.get('$(PACKAGE_NAME)', ''))")
120-
pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}" --find-links="${TORCHVISION_URL}"
119+
pip install -e ".[${extra}dev]" pytest-timeout -U --extra-index-url="${TORCH_URL}"
121120
pip install setuptools==75.6.0 jsonargparse==4.35.0
122121
displayName: "Install package & dependencies"
123122

0 commit comments

Comments
 (0)