CaoWGG / TensorRT-YOLOv4

tensorrt5, yolov4, yolov3,yolov3-tniy,yolov3-tniy-prn
265 stars 76 forks source link

mContext->execute 报错 #26

Open konglingzheng opened 4 years ago

konglingzheng commented 4 years ago

我用的是jetson nano 自带Tensorrt 6.0 cuda 10.0 按照您的步骤运行到此步骤 ./runDet -i model/yolov3.engine -p dog.jpg -v nuscenes_mini.mp4 报错信息为:terminate called after throwing an instance of 'thrust::system::system_error' what(): parallel_for failed: no kernel image is available for execution on the device 报错的位置在 trt.cpp : mContext->execute(1,&mCudaBuffers[0]); 大概查了一下,大概是cuda版本问题。 想知道的是这个程序,如何可以在我这个环境下运行,在不更改cuda和tensorrt版本的情况下 还有个疑问: 在TensotRT-YOLOv4目录下的CMakeLists.txt文件中, set(GPU_ARCHS 62) set(TENSORRT_ROOT /usr/local/TensorRT-5.0.2.6) 这两句起到什么作用。

期待您的解答与交流。感谢

CaoWGG commented 4 years ago

@konglingzheng 更改onnx-tensorrt/yolo.cu,src/resize.cu的blocksize为256. 更改https://github.com/CaoWGG/TensorRT-YOLOv4/blob/4d7c2edce99e8794a4cb4ea3540d51ce91158a36/onnx-tensorrt/FancyActivation.cu#L60 的leakrelu的实现方式(把blocksize写小了),参考darknet的https://github.com/AlexeyAB/darknet/blob/e08a818d5fd12f2f6e6e262267b132be195b40a5/src/activation_kernels.cu#L452