TensorRt undefined symbol: getBuilderPluginRegistry

Description

After fitting YoloX model I am trying to convert it to engine, so I can run it using tensor cores.

To do so I cloned torch2trt that requires tensorrt library.
I installed tensorrt via pip wheel in python3.8 venv:

  • pip install --upgrade setuptools pip
  • pip install nvidia-pyindex
  • pip install --upgrade nvidia-tensorrt

When I run python setup.py install I get ImportError:

Traceback (most recent call last):
  File "setup.py", line 2, in <module>
    import tensorrt
  File "/mnt/dev/YOLOX/venv/lib/python3.8/site-packages/tensorrt/__init__.py", line 67, in <module>
    from .tensorrt import *
ImportError: /mnt/dev/YOLOX/venv/lib/python3.8/site-packages/tensorrt/libnvonnxparser.so.8: undefined symbol: getBuilderPluginRegistry

Environment

TensorRT Version: 8.2.3-1
GPU Type: NVIDIA GeForce RTX 3060
Nvidia Driver Version: 495.29.05
CUDA Version: 11.3
CUDNN Version: 8
Operating System + Version: Ubuntu 20.04.4 LTS
Python Version (if applicable): 3.8
PyTorch Version (if applicable): 1.11.0+cu113

Had to downgrade to version 8.0.1.6
pip install --upgrade nvidia-tensorrt==8.0.1.6

3 Likes

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.