TensorRT Algorithm selector

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

Description

The document for the TensorRT Algorithm selector using Python API is not clear.Can anyone please help me to implement an algorithm selector in TensorRT with an example code snippet?

Environment

TensorRT Version : 8.0
GPU Type :
Nvidia Driver Version : 470.57.02
CUDA Version : 11.3
CUDNN Version : 8.2
Operating System + Version : ubuntu20.04
Python Version (if applicable) : 3.8
TensorFlow Version (if applicable) : 2.5
PyTorch Version (if applicable) :
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,
Please check the below link, as they might answer your concerns

Thanks!

4

Hello, thanks for your response.
I would like to have an example code using a different algorithm selector using python API.

The documentation for TRT Algorithm selector is not clear for implementation.

https://docs.nvidia.com/deeplearning/tensorrt/api/python_api/infer/AlgorithmSelector/pyAlgorithmSelector.html

5

Hi,

Looks like currently we do not have a python example, we have c++ example here,

Thank you.