pytorch / pytorch/TensorRT

❓ [Question] Model inference is much slower after updating to TensorRT 9.3

Open
#2,822 3 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

question story: Performance & Benchmarking
Dominant language
Python
Stars
3k
Forks
410
Avg merge
3d 18h
Merged PRs (30d)
78

Description

❓ Question

I have a VIT model for object detection. The model inference speed in the tensort 8.5 environment is 190ms per frame. However when I updated to TensorRT 9.3, Inference slowed down to 250ms per frame.

I acquired the C++ dynamic library by compiling the latest Torch-TensorRT source code.

What might be causing this issue?

Environment

Build information about Torch-TensorRT can be found by turning on debug messages

  • Libtorch Version (e.g., 1.0): 2.2.1
  • CPU Architecture:
  • OS (e.g., Linux): ubuntu22.04
  • How you installed PyTorch (conda, pip, libtorch, source):
  • Build command you used (if compiling from source):
  • Are you using local sources or building from archives: Yes
  • Python version:
  • CUDA version: 12.2
  • GPU models and configuration:
  • Any other relevant information:

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

No source file or test is identified. Start by reviewing the reported Torch-TensorRT build context and comparing the VIT inference measurements between TensorRT 8.5 and 9.3, filling in the missing GPU, build, installation, and CPU details. Done means the slowdown has a reproducible configuration and an identified cause.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp, pytorch
Domain
machine-learning, performance
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.