BiSeNet icon indicating copy to clipboard operation
BiSeNet copied to clipboard

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

Open upupbo opened this issue 11 months ago • 2 comments
trafficstars

/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 请问这个应该如何解决

upupbo avatar Dec 12 '24 08:12 upupbo

Hi,

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

CoinCheung avatar Dec 19 '24 12:12 CoinCheung

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

upupbo avatar Dec 24 '24 09:12 upupbo