Closed
Description
Bug Description
Installing the torch_tensorrt package version 1.4.0 from pip with pip install torch_tensorrt==1.4.0
does not install the required tensorrt
package.
To Reproduce
Use pip install torch_tensorrt==1.4.0
Run this python line:
import tensorrt
Expected behavior
TensorRT should be installed as a dependecy of this package.
Environment
Build information about Torch-TensorRT can be found by turning on debug messages
- Torch-TensorRT Version (e.g. 1.0.0): 1.4.0
- PyTorch Version (e.g. 1.0): 2.0.1
- CPU Architecture: x86
- OS (e.g., Linux): Linux
- How you installed PyTorch (
conda
,pip
,libtorch
, source): installed by torch_tensorrt - Build command you used (if compiling from source):
- Are you using local sources or building from archives:
- Python version: 3.10
- CUDA version:
- GPU models and configuration:
- Any other relevant information:
Additional context
The 1.3.0 wheels contain this line in the METADATA files:
Requires-Dist: torch (<1.14.0,>=1.13.0)
Requires-Dist: tensorrt (<8.6.0,>=8.5.1.7)
The 1.4.0 wheels contain only
Requires-Dist: torch (<2.1,>=2.0.1)