FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎
GNU General Public License v2.0
1.25k stars 195 forks source link

jetson nano B01推理失败 #134

Closed teacher1314 closed 1 week ago

teacher1314 commented 1 week ago

使用的CUDA是10.2 CUDNN是8.2.1 tensorrt是8.2.1 报这个错误: terminate called after throwing an instance of 'thrust::system::system_error' what(): radix_sort: failed on 2nd step: cudaErrorInvalidConfiguration: invalid configuration argument 然后我定位到报错函数的函数是tensorrt-alpha-ros/src/utils/kernel_function.cu下的nmsDeviceV2函数,这个函数里面的是thrust::sort_by_key(thrust::device, p_conf, p_conf + srcHeight, p_idx, thrust::greater());这行代码问题