search

linghu8812 / tensorrt_inference

696 stars 205 forks source link

Segmentation fault #170

Closed Egorundel closed 1 year ago

Egorundel commented 1 year ago

Hello, I have the following problem:

Screenshot from 2023-07-06 12-12-10

Can you help me to solve this problem, please?

TensorRT Version: 7.2.2.3.

NVIDIA GPU: RTX 2070

CUDA Version: 11.1

Operationg System: Ubuntu 20.04

Egorundel commented 1 year ago

The problem was solved by replacing void *buffers[2]; in model.h on void *buffers[5];

and in model.cpp changed the line assert(engine->getNbBindings() == 2); to assert(engine->getNbBindings() == 5);

Do this if you use the NMS Plugin in the model :)