diff --git a/utilities.py b/utilities.py index 1ff6499..5b922c1 100644 --- a/utilities.py +++ b/utilities.py @@ -234,6 +234,7 @@ def build( update_output_names=None, ): print(f"Building TensorRT engine for {onnx_path}: {self.engine_path}") + trt.init_libnvinfer_plugins(None, "") p = [Profile()] if input_profile: p = [Profile() for i in range(len(input_profile))]