search

enazoe / yolo-tensorrt

TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is useful to you,please Star it.
MIT License
1.18k stars 313 forks source link

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

Open shiguoliang opened 3 years ago

shiguoliang commented 3 years ago

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

enazoe commented 3 years ago

@shiguoliang which version of your trt ?

shiguoliang commented 3 years ago

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

enazoe commented 3 years ago

@shiguoliang use trt7.1 and recommend version

shiguoliang commented 3 years ago

thanks very much

yutao007 commented 3 years ago

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