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

cuda12.2,tensorrt8.6,onnx转tensorrt报错 #339

Open
upupbo opened this issue Dec 12, 2024 · 2 comments
Open

cuda12.2,tensorrt8.6,onnx转tensorrt报错 #339

upupbo opened this issue Dec 12, 2024 · 2 comments

Comments

@upupbo
Copy link

upupbo commented Dec 12, 2024

/home/c24702/code/C++_project/bisenet1/BiSeNet/tensorrt/trt_dep.cpp: In member function 'void SemanticSegmentTrt::parse_to_engine(std::string, std::string, std::string, std::string)':
/home/c24702/code/C++_project/bisenet1/BiSeNet/tensorrt/trt_dep.cpp:106:48: error: 'kBF16' is not a member of 'nvinfer1::BuilderFlag'
106 | config->setFlag(nvinfer1::BuilderFlag::kBF16);
| ^~~~~
/home/c24702/code/C++_project/bisenet1/BiSeNet/tensorrt/trt_dep.cpp: In member function 'std::vector SemanticSegmentTrt::inference(std::vector&)':
/home/c24702/code/C++_project/bisenet1/BiSeNet/tensorrt/trt_dep.cpp:229:14: error: 'class nvinfer1::IExecutionContext' has no member named 'setOutputTensorAddress'; did you mean 'getOutputTensorAddress'?
229 | context->setOutputTensorAddress(output_name.c_str(), buffs[1]);
| ^~~~~~~~~~~~~~~~~~~~~~
| getOutputTensorAddress
make[2]: *** [CMakeFiles/segment.dir/build.make:90: CMakeFiles/segment.dir/trt_dep.cpp.o] Error 1
make[1]: *** [CMakeFiles/Makefile2:111: CMakeFiles/segment.dir/all] Error 2
make: *** [Makefile:91: all] Error 2
请问这个应该如何解决

@CoinCheung
Copy link
Owner

Hi,

Please use tensorrt-10. The code is updated to tensorrt-10.

@upupbo
Copy link
Author

upupbo commented Dec 24, 2024

Our project needs to use tensorrt8.6. Can you use tensorrt8.6 to help implement a version?

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

2 participants