TensorRT
TensorRT copied to clipboard
🐛 [Bug] Encountered bug when using Torch-TensorRT
Bug Description
Presence of BUILD file in released wheel does break use of package in Bazel with rules-python (https://github.com/bazel-contrib/rules_python/issues/2780).
To Reproduce
Steps to reproduce the behavior:
- See here: https://github.com/scanvio/bazel-torch-tensorrt
Expected behavior
Import works without any issues.
Environment
Build information about Torch-TensorRT can be found by turning on debug messages
- Torch-TensorRT Version (e.g. 1.0.0): 2.6.0, CUDA 12.6
- PyTorch Version (e.g. 1.0): 2.6.0
- CPU Architecture: x86/64
- OS (e.g., Linux): Ubuntu 24.04
- How you installed PyTorch (
conda,pip,libtorch, source): Bazel, rules-python - Build command you used (if compiling from source): -
- Are you using local sources or building from archives: -
- Python version: 3.11
- CUDA version: 12.6
- GPU models and configuration: Tesla T4
- Any other relevant information: -
Additional context
@apbose for DLFW CI, do we still need this build file? or can we pull it in as external to the wheel?
Thanks for the PR @mering. The above PR breaks our internal CI for the torchTRT cpp tests, working on the workaround for that.