File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 9191 fi
9292
9393 # install the base so we can adjust other packages
94- uv pip install . --find-links ="${TORCH_URL}"
94+ uv pip install . --extra-index-url ="${TORCH_URL}"
9595 echo "Adjust torch versions in requirements files"
9696 PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])")
9797 uv pip install wget packaging
@@ -112,7 +112,7 @@ run: |
112112 fi
113113
114114 extra=$(python -c "print({'lightning': 'fabric-'}.get('$(PACKAGE_NAME)', ''))")
115- uv pip install ".[${extra}dev]" --upgrade --find-links ="${TORCH_URL}"
115+ uv pip install ".[${extra}dev]" --upgrade --extra-index-url ="${TORCH_URL}"
116116
117117 python requirements/collect_env_details.py
118118 python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu >= 2, f'GPU: {mgpu}'"
You can’t perform that action at this time.
0 commit comments