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