yolo-tensorrt icon indicating copy to clipboard operation
yolo-tensorrt copied to clipboard

it is failed for Yolov4 transfer to tensorrt! Please Help answer

Open shiguoliang opened this issue 4 years ago • 5 comments

Thank you very much for sharing the project, but I have a error transferring to yolov4. It is show mistake "Internal error: could not find any implementation for node (Unnamed Layer* 271) [Deconvolution], try increasing the workspace size with IBuilder:: setMaxWorkspaceSize() ../builder/tacticOptimizer.cpp (1523) - OutOfMemory Error in computeCosts: 0", Please help me to answer, thanks very much

shiguoliang avatar Sep 19 '20 23:09 shiguoliang

@shiguoliang which version of your trt ?

enazoe avatar Sep 20 '20 05:09 enazoe

ubuntu16,TensorRT-7.0.0.11 , cuda 9.0 , cudnn 7.6.3 , opencv4.1 ;

shiguoliang avatar Sep 20 '20 09:09 shiguoliang

@shiguoliang use trt7.1 and recommend version

enazoe avatar Sep 20 '20 10:09 enazoe

thanks very much

shiguoliang avatar Sep 20 '20 11:09 shiguoliang

setMaxWorkspaceSize()将这个改成1<<30 试试

yutao007 avatar Sep 28 '20 09:09 yutao007