Skip to content

Commit 53265fb

Browse files
committed
add index url for torch tensorrt
1 parent 986077c commit 53265fb

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

.lightning/workflows/pytorch.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,7 @@ run: |
125125
# Use find-links to prefer CUDA-specific packages from PyTorch index
126126
uv pip install -e ".[${extra}dev]" --upgrade \
127127
--find-links="https://download.pytorch.org/whl/${UV_TORCH_BACKEND}"
128+
--find-links="https://download.pytorch.org/whl/${UV_TORCH_BACKEND}/torch-tensorrt"
128129
uv pip list
129130
130131
echo "Ensure only a single package is installed"

0 commit comments

Comments
 (0)