chenzhi1992 / TensorRT-SSD

Use TensorRT API to implement Caffe-SSD, SSD(channel pruning), Mobilenet-SSD
251 stars 84 forks source link

cuda error in allocate: 2 #43

Open ujsyehao opened 5 years ago

ujsyehao commented 5 years ago

I modify main.cpp to support for video detection, it has the error: runtime.cpp (16) - Cuda Error in allocate: 2 terminate called after throwing an instance of 'nvinfer1::CudaError' I search the problem, Most of the explains are cuda memory leak. Can you give me some advice?