Jetson orin nx ModuleNotFoundError: No module named 'tensorrt'

[bragiwang]

Hi there,
I burnout the latest jetpack 6.1 through sdk manager, but skip the sdk components. when I run .engine file, the console said

Now i manually install sdk components through

sudo apt update
sudo apt install nvidia-jetpack

but the installed version is 10.3+cuda 12.5.

Since i have cuda 12.6 installed with jetpack 6.1, I still can’t import tensorrt, the error message remain the same.

also tried

sudo apt apt-get install python3-libnvinfer*

all packages are correctly installed.

is this a bug or? any solutions are appreciate.

[AastaLLL]

Hi,

The TensorRT python package should be installed with the command you shared.
Do you use a virtual environment?

If yes, could you try if the import works outside of the environment?

Thanks.

[AastaLLL]

Hi,

The version is compatible to the JetPack 6.1 environment.

$ dpkg -l | grep tensorrt
ii  nvidia-tensorrt                              6.1+b123                                          arm64        NVIDIA TensorRT Meta Package
ii  nvidia-tensorrt-dev                          6.1+b123                                          arm64        NVIDIA TensorRT dev Meta Package
ii  tensorrt                                     10.3.0.30-1+cuda12.5                              arm64        Meta package for TensorRT
ii  tensorrt-libs                                10.3.0.30-1+cuda12.5                              arm64        Meta package for TensorRT runtime libraries

$ python3 
Python 3.10.12 (main, Sep 11 2024, 15:47:36) [GCC 11.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorrt
>>> tensorrt.__version__
'10.3.0'

Based on your log, it looks like you are using a virtual environment.
Could you create it with --system-site-packages to see if it works?

Thanks.