Can't install tensorrt package python

AI & Data Science Deep Learning (Training & Inference) TensorRT
,
1

Description

So basically i wanted to install
pip install nvidia-pyindex
pip install nvidia-tensorrt
packages to export data from .pt to .engine using yolov5 but it returns this :

Collecting nvidia-tensorrt
  Downloading nvidia-tensorrt-0.0.1.dev5.tar.gz (7.9 kB)
  Preparing metadata (setup.py) ... error
  error: subprocess-exited-with-error

  × python setup.py egg_info did not run successfully.
  │ exit code: 1
  ╰─> [17 lines of output]
      Traceback (most recent call last):
        File "<string>", line 2, in <module>
        File "<pip-setuptools-caller>", line 34, in <module>
        File "C:\Users\hohom\AppData\Local\Temp\pip-install-zsfwwk16\nvidia-tensorrt_7886ed69e8de427b9749ffd8f7c4dc30\setup.py", line 150, in <module>
          raise RuntimeError(open("ERROR.txt", "r").read())
      RuntimeError:
      ###########################################################################################
      The package you are trying to install is only a placeholder project on PyPI.org repository.
      This package is hosted on NVIDIA Python Package Index.

      This package can be installed as:
      ```
      $ pip install nvidia-pyindex
      $ pip install nvidia-tensorrt
      ```
      ###########################################################################################

      [end of output]

  note: This error originates from a subprocess, and is likely not a problem with pip.
error: metadata-generation-failed

× Encountered error while generating package metadata.
╰─> See above for output.

note: This is an issue with the package mentioned above, not pip.
hint: See above for details.

P.S I followed all the instructions on official installation guide

Environment

TensorRT Version: 8.6
GPU Type: RTX 3090 TI
Nvidia Driver Version: 537.58
CUDA Version: 12.1
CUDNN Version: 8.9.0.131
Operating System + Version: Windows 11 22H2
Python Version (if applicable): 3.9
TensorFlow Version (if applicable):
PyTorch Version (if applicable): 2.1
Baremetal or Container (if container which image + tag):

Relevant Files

Please attach or include links to any models, data, files, or scripts necessary to reproduce your issue. (Github repo, Google Drive, Dropbox, etc.)

Steps To Reproduce

Please include:

  • Exact steps/commands to build your repro
  • Exact steps/commands to run your repro
  • Full traceback of errors encountered
2

Hi,
Can you try running your model with trtexec command, and share the “”–verbose"" log in case if the issue persist

You can refer below link for all the supported operators list, in case any operator is not supported you need to create a custom plugin to support that operation

Also, request you to share your model and script if not shared already so that we can help you better.

Meanwhile, for some common errors and queries please refer to below link:

Thanks!

3

log.txt (6.9 MB) --verbose output

https://drive.google.com/file/d/1AQ9wJ_O3eomayxRts6e3Z7mr8LkCRn11/view?usp=sharing - onnx model

4

Hi @Pudge228 ,
Can you please try installing TRT from the below link and let us know if problem still persist?

Thanks

1 Like
5

@Pudge228 Did u find a solution ?
I have the same issue with my AGX…

6

‘3.2.4. Zip File Installation’ works OK, on Windows10 + Anaconda3 (Python3.11) environment.