TensorRT-Alpha icon indicating copy to clipboard operation
TensorRT-Alpha copied to clipboard

YOLOv8目标检测,多线程核心转储的问题

Open dcstempt opened this issue 1 year ago • 0 comments

当我在线程A:初始化YOLOv8,设置显卡的id,并随机生成一张RGB图像进行模型预热; 线程A有以下一个重要步骤: cudaStatus = cudaSetDevice(this->gpu_id); // 设置显卡id this->yolo = std::make_shared<YOLOV8>(this->param); // 调用构造函数 yolo->init(this->trt_file) // 调用init函数 this->yolo->check();

此时,我在线程B,调用YOLOv8进行图像检测,将会报错: [E] [TRT] 1: [caskUtils.cpp::createCaskHardwareInfo::852] Error Code 1: Cuda Runtime (an illegal memory access was encountered)

但是很奇怪的一点是,如果显卡id=0,多线程不会出现问题。只有当gpu_id != 0是,才会报错

dcstempt avatar Dec 23 '23 02:12 dcstempt