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 986077c commit 53265fbCopy full SHA for 53265fb
.lightning/workflows/pytorch.yml
@@ -125,6 +125,7 @@ run: |
125
# Use find-links to prefer CUDA-specific packages from PyTorch index
126
uv pip install -e ".[${extra}dev]" --upgrade \
127
--find-links="https://download.pytorch.org/whl/${UV_TORCH_BACKEND}"
128
+ --find-links="https://download.pytorch.org/whl/${UV_TORCH_BACKEND}/torch-tensorrt"
129
uv pip list
130
131
echo "Ensure only a single package is installed"
0 commit comments