Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Error with trt 10.x #5

Open
mohamedsamirx opened this issue Oct 28, 2024 · 0 comments
Open

Error with trt 10.x #5

mohamedsamirx opened this issue Oct 28, 2024 · 0 comments

Comments

@mohamedsamirx
Copy link

got an error using tensorrt version 10.x and i solve it by adding
"const char* input_name = engine->getIOTensorName(0);

// retrieves the name of the output tensor (e.g., the detection results).
const char* output_name = engine->getIOTensorName(1);

// Set the input tensor address
// Setting these addresses ensures that TensorRT knows where to read the input data and write the output data.
// assigns memory addresses (in GPU buffers) for both input and output tensors.

// This buffer contains the preprocessed input image (in GPU memory).
context->setTensorAddress(input_name, gpu_buffers[0]);

// This buffer will store the inference results (detections).
context->setTensorAddress(output_name, gpu_buffers[1]);
"
you may check it. and thank you again

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant