-
Notifications
You must be signed in to change notification settings - Fork 681
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
Can not run with another TensorRT Engine in one program. #373
Comments
Hi Water2style, Thanks for reaching out! Hmm. I haven't seen this before personally. Do you mind sharing the code you used to produce this issue? Also, the following information may also be helpful
Best, |
Sorry i cannot share the code, but it just a strange try(tensoRT engine + trt.pth) , ocuurs : Cudnn Error configure:7 (CUDNN_STATUS_MAPPING_ERROR) and CUDNN_STATUS_EXECUTION_FAILED |
I encountered the same problem.when I use torch2trt and another TensorRT engine, an illegal memory access was encountered,but they can run correctly separately。 |
@BarryKCL excuse me, have you solved it? I'm facing the same problem when running multi tensorRT_engines in one program. |
Have anyone solved it? I'm facing the same problem when running a tensorRT_engine (yolov5) and another classification network (resnet_trt.pth) in the same program. Thanks! |
@fabiozappo yes, I was able to solve it bro. All you need to do is just a simple trick: |
it works! |
@xuatpham @fabiozappo |
before loading the yolo, just initialize a tensor good luck bro |
I got a good model_trt by torch2trt, and it can run alone in a program.
But when i add an TensorRT engine in the program, it seems that program can not run an TensorRT engine and a model_trt together. But they can run correctly separately.
If I want to run two models in one program, what should i do? Thanks!
The text was updated successfully, but these errors were encountered: