ModuleNotFoundError: No module named 'tensorrt'

Robotics & Edge Computing Jetson & Embedded Systems Jetson Nano
1

development environment:conda+python3.6+cuda10.2
I want to run trt_pose on this nano platform. when import torch2trt, there is an error:ModuleNotFoundError: No module named ‘tensorrt’.
i need help.

3

Hi,

Have you installed TensorRT in your environment?
The package is one of the components listed in the JetPack.

Thanks.

4

how i can install TensorRT in my environment.
i have download the package nv-tensorrt-repo-ubuntu1804-cuda10.2-trt6.0.1.8-ga-20191108_1-1_amd64.deb. but the package is not applicable to arm64. Where should I download it.

1 Like
5

Hi,

Please use JetPack (from SDKmanager) to get the TensorRT for Jetson.

Thanks.

6

usern:~$ sudo jetson_release

  • NVIDIA Jetson Nano (Developer Kit Version)
    • Jetpack 4.4.1 [L4T 32.4.4]
    • NV Power Mode: MAXN - Type: 0
    • jetson_stats.service: active
  • Libraries:
    • CUDA: 10.2.89
    • cuDNN: 8.0.0.180
    • TensorRT: 7.1.3.0
    • Visionworks: NOT_INSTALLED
    • OpenCV: 4.1.1 compiled CUDA: NO
    • VPI: 4.4.1-b50
    • Vulkan: 1.2.70
      now, i can see this information, it’s right?
7

Hi,

Based on the log, your device is using JetPack4.4.1.
But somehow, the VisionWorks installation is failed.

To ensure the TensorRT functionality, could you run the below command to see if all the packages are well-installed?
Especially, the python3-libnvinfer and python3-libnvinfer-dev?

$ dpkg -l | grep nvinfer

ii  libnvinfer-bin                                7.1.3-1+cuda10.2                                 arm64        TensorRT binaries
ii  libnvinfer-dev                                7.1.3-1+cuda10.2                                 arm64        TensorRT development libraries and headers
ii  libnvinfer-doc                                7.1.3-1+cuda10.2                                 all          TensorRT documentation
ii  libnvinfer-plugin-dev                         7.1.3-1+cuda10.2                                 arm64        TensorRT plugin libraries
ii  libnvinfer-plugin7                            7.1.3-1+cuda10.2                                 arm64        TensorRT plugin libraries
ii  libnvinfer-samples                            7.1.3-1+cuda10.2                                 all          TensorRT samples
ii  libnvinfer7                                   7.1.3-1+cuda10.2                                 arm64        TensorRT runtime libraries
ii  python-libnvinfer                             7.1.3-1+cuda10.2                                 arm64        Python bindings for TensorRT
ii  python-libnvinfer-dev                         7.1.3-1+cuda10.2                                 arm64        Python development package for TensorRT
ii  python3-libnvinfer                            7.1.3-1+cuda10.2                                 arm64        Python 3 bindings for TensorRT
ii  python3-libnvinfer-dev                        7.1.3-1+cuda10.2                                 arm64        Python 3 development package for TensorRT

Thanks

9

ii libnvinfer-bin 7.1.3-1+cuda10.2 arm64 TensorRT binaries
ii libnvinfer-dev 7.1.3-1+cuda10.2 arm64 TensorRT development libraries and headers
ii libnvinfer-doc 7.1.3-1+cuda10.2 all TensorRT documentation
ii libnvinfer-plugin-dev 7.1.3-1+cuda10.2 arm64 TensorRT plugin libraries
ii libnvinfer-plugin7 7.1.3-1+cuda10.2 arm64 TensorRT plugin libraries
ii libnvinfer-samples 7.1.3-1+cuda10.2 all TensorRT samples
ii libnvinfer7 7.1.3-1+cuda10.2 arm64 TensorRT runtime libraries
ii python-libnvinfer 7.1.3-1+cuda10.2 arm64 Python bindings for TensorRT
ii python-libnvinfer-dev 7.1.3-1+cuda10.2 arm64 Python development package for TensorRT
ii python3-libnvinfer 7.1.3-1+cuda10.2 arm64 Python 3 bindings for TensorRT
ii python3-libnvinfer-dev 7.1.3-1+cuda10.2 arm64 Python 3 development package for TensorR

But still have the problem :
no module named “tensorrt”

10

Hi 921825229,

Please help to open a new topic for your issue. Thanks