Toybrick

关于文档里面支持的跑多核npu时出现的问题

roy

新手上路

积分
36
楼主
发表于 2023-4-7 14:38:45    查看: 2171|回复: 3 | [复制链接]    打印 | 只看该作者
运行代码:
    rknn_core_mask core_mask = RKNN_NPU_CORE_0_1_2;
    model = load_model(model_path, &model_len);
    ret = rknn_init(&ctx, model, model_len, RKNN_FLAG_MEM_ALLOC_OUTSIDE, NULL);
    int eeee = rknn_set_core_mask(ctx, core_mask);


报错:
E RKNN: [06:34:40.715] Not support core mask: 7, fallback to single core auto mode
E RKNN: [06:34:40.715] NN Compiler/Model Version is 1.3.0 now
E RKNN: [06:34:40.715] rknn_set_core_mask: failed to set core mask: 7

环境:
RK3588 Debian11 用的最新的固件5.10.110和自己编译的5.10.66都试过
rknn的so库是从github上拉的,github上的readme已经写到了release 1.4,但报错里面显示1.3,是不是库还是有问题?
想用3核对一个模型进行卷积加速优化的,卷积网络的权重用的一层高斯核测试的。

回复

使用道具 举报

roy

新手上路

积分
36
沙发
 楼主| 发表于 2023-4-7 14:42:03 | 只看该作者
本帖最后由 roy 于 2023-4-7 15:57 编辑

补充:内核一:rknn_api/rknnrt version: 1.4.0 (a10f100eb@2022-09-09T09:07:14), driver version: 0.6.4内核二:rknn_api/rknnrt version: 1.4.0 (a10f100eb@2022-09-09T09:07:14), driver version: 0.8.2

均报下面错误
E RKNN: [07:56:04.676] Not support core mask: 7, fallback to single core auto mode
E RKNN: [07:56:04.676] NN Compiler/Model Version is 1.2.6 now
E RKNN: [07:56:04.676] rknn_set_core_mask: failed to set core mask: 7

回复

使用道具 举报

wudiking

高级会员

积分
689
板凳
发表于 2023-4-10 06:39:33 | 只看该作者
roy 发表于 2023-4-7 14:42
补充:内核一:rknn_api/rknnrt version: 1.4.0 (a10f100eb@2022-09-09T09:07:14), driver version: 0.6.4 ...

把rknn的模型用1.4.0后的版本重新转换一下,我之前用过老模型就是这个问题,重新转一下就好了
回复

使用道具 举报

roy

新手上路

积分
36
地板
 楼主| 发表于 2023-4-11 11:20:38 | 只看该作者
wudiking 发表于 2023-4-10 06:39
把rknn的模型用1.4.0后的版本重新转换一下,我之前用过老模型就是这个问题,重新转一下就好了 ...

感谢!已经解决,确实是这样
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表