Skip to content

Commit ae297fe

Browse files
committed
ci: resolve torch-tensorrt from PyPI to keep the TorchScript frontend
The 2.13.0+cu130 wheel on download.pytorch.org is pure Python (608 KB, zero .so files); 2.12.1+cu130 bundled libtorchtrt.so. Since a local version sorts above the plain release, the find-links entry always won and the TorchScript frontend went missing. PyPI's 2.13.0 ships the libraries and is likewise a CUDA 13 build.
1 parent a90a439 commit ae297fe

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

.lightning/workflows/pytorch.yml

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -122,10 +122,11 @@ run: |
122122
echo "Install package"
123123
extra=$(python -c "print({'lightning': 'pytorch-'}.get('${PACKAGE_NAME}', ''))")
124124
125-
# Use find-links to prefer CUDA-specific packages from PyTorch index
125+
# Prefer CUDA-specific builds from the PyTorch index.
126+
# torch-tensorrt is intentionally left to PyPI: since 2.13.0 its `+cuXXX` wheels no longer
127+
# bundle `libtorchtrt.so`, which silently disables the TorchScript frontend.
126128
uv pip install -e ".[${extra}dev,${extra}test_gpu]" --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+
--find-links="https://download.pytorch.org/whl/${UV_TORCH_BACKEND}"
129130
uv pip list
130131
131132
echo "Ensure only a single package is installed"

0 commit comments

Comments
 (0)