-
Notifications
You must be signed in to change notification settings - Fork 2.3k
Description
Description
Fresh install of pip install tensorrt==10.2.0
Following engine build crashes on Ubuntu 22.04.4 LTS:
from polygraphy.backend.trt import EngineFromNetwork
EngineFromNetwork(
network,
config=CreateConfig(fp16=fp16,
tf32=tf32,
int8=int8,
refittable=enable_refit,
profiles=[p],
load_timing_cache=timing_cache,
builder_optimization_level=3,
**extra_build_args
),
save_timing_cache=timing_cache
)()
Error message:
IBuilder::buildSerializedNetwork: Error Code 6: API Usage Error (Unable to load library: libnvinfer_builder_resource_win.so.10.2.0: libnvinfer_builder_resource_win.so.10.2.0: cannot open shared object file: No such file or directory)
Build works fine on 10.1.0 and 10.0.0
Environment
TensorRT Version: 10.2.0
NVIDIA GPU: RTX 4090
NVIDIA Driver Version: 550
CUDA Version: 12.1.r12.1
CUDNN Version: 8.9.7
Operating System: Ubuntu 22.04.4 LTS
Python Version (if applicable):
PyTorch Version (if applicable): 2.3.1
Baremetal or Container (if so, version):
Relevant Files
Model link:
Steps To Reproduce
Commands or scripts:
Have you tried the latest release?:
This is the latest release.
Can this model run on other frameworks? For example run ONNX model with ONNXRuntime (polygraphy run <model.onnx> --onnxrt
):
Yes, the above command completes successfully, the ONNX file is correct.