Skip to content

Commit 4674012

Browse files
committed
add find link
1 parent 98a70c9 commit 4674012

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

.lightning/workflows/pytorch.yml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,9 @@ run: |
123123
extra=$(python -c "print({'lightning': 'pytorch-'}.get('${PACKAGE_NAME}', ''))")
124124
125125
# Use find-links to prefer CUDA-specific packages from PyTorch index
126-
uv pip install -e ".[${extra}dev]" --upgrade
126+
uv pip install -e ".[${extra}dev]" --upgrade \
127+
--find-links="https://download.pytorch.org/whl/${UV_TORCH_BACKEND}"
128+
uv pip list
127129
128130
echo "Ensure only a single package is installed"
129131
if [ "${PACKAGE_NAME}" == "pytorch" ]; then

0 commit comments

Comments
 (0)