fix: prefer CUDA-specific packages from PyTorch index using find-links

Author: bhimrazy

.lightning/workflows/pytorch.yml

@@ -121,7 +121,10 @@ run: |
 
   echo "Install package"
   extra=$(python -c "print({'lightning': 'pytorch-'}.get('${PACKAGE_NAME}', ''))")
-  uv pip install -e ".[${extra}dev]" --upgrade --extra-index-url https://download.pytorch.org/whl/cu${CUDA_VERSION_MM}
+
+  # Use find-links to prefer CUDA-specific packages from PyTorch index
+  uv pip install -e ".[${extra}dev]" --upgrade \
+    --find-links https://download.pytorch.org/whl/cu${CUDA_VERSION_MM}
 
   echo "Ensure only a single package is installed"
   if [ "${PACKAGE_NAME}" == "pytorch" ]; then