ModuleNotFoundError: No module named 'tensorrt'

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.

Hi,

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

Thanks.

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.

Hi,

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

Thanks.

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?

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