We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 98a70c9 commit 4674012Copy full SHA for 4674012
.lightning/workflows/pytorch.yml
@@ -123,7 +123,9 @@ run: |
123
extra=$(python -c "print({'lightning': 'pytorch-'}.get('${PACKAGE_NAME}', ''))")
124
125
# Use find-links to prefer CUDA-specific packages from PyTorch index
126
- uv pip install -e ".[${extra}dev]" --upgrade
+ uv pip install -e ".[${extra}dev]" --upgrade \
127
+ --find-links="https://download.pytorch.org/whl/${UV_TORCH_BACKEND}"
128
+ uv pip list
129
130
echo "Ensure only a single package is installed"
131
if [ "${PACKAGE_NAME}" == "pytorch" ]; then
0 commit comments