【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);
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值