tensorrt engine->serialize()报错调试

1.打开log信息打印

/******************************/
// TensorRT utility
/******************************/
class Logger : public ILogger
{
    void log(Severity severity, const char *msg) override
    {
        if (severity != Severity::kINFO)
            std::cout << msg << std::endl;
    }
};

2.查看如下报错

cuda/cudaFusedConvActLayer.cpp (287) - Cuda Error in executeFused: 9
cuda/cudaFusedConvActLayer.cpp (287) - Cuda Error in executeFused: 9

可在gedit /usr/local/cuda/include/driver_types.h中找到相关报错信息

3.得到结论

一维卷积单个维度太大,更换为二维卷积

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值