【tensorrt】——全局静态变量释放cudastream时，报driver shutting down问题?

声明一个全局类，在类的析构函数中会释放显存。

yolodet::yoloNet net; int main(int argc, char* argv[]){ std::string engine = "model/yolov4.engine"; net.loadmodel(engine); return 0; }

报错： CUDA error driver shutting down at /home/data/CM/3_image_classification/pytorch_image_classification/device/tensorrt2/src/trt.cpp:195runDet: /home/data/CM/3_image_classification/pytorch_image_classification/device/tensorrt2/src/trt.cpp:195: yolodet::yoloNet::~yoloNet(): Assertion `0’ failed. Aborted (core dumped)

相似的问题： http://bbs.gpuworld.cn/index.php?topic=59560.0

解决方案 直接初始化类，不要先声明，再初始化。

// 先声明再初始化 // yolodet::yoloNet net; std::string engine = "output/mobilenetv2.engine"; yolodet::yoloNet net(engine);

问题二

这样声明全局类还是会报错，只能在用new了。

std::string engine = "output/mobilenetv2.engine"; static yolodet::yoloNet net(engine);

解决方案

std::string engine = "output/mobilenetv2.engine"; yolodet::yoloNet* net = new yolodet::yoloNet(engine);