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.19k stars 316 forks source link

i ha #122

Open henbucuoshanghai opened 3 years ago

henbucuoshanghai commented 3 years ago

cudaMemcpyAsync(out1, buffers[1], bufferSize[1], cudaMemcpyDeviceToHost, stream);

malloc(): memory corruption Aborted (core dumped)

henbucuoshanghai commented 3 years ago

when i have many outputs,like two or three? how to get it from the gpu?