Toybrick

init_runtime()过程每次都要花费很多时间?

troy

版主

积分
2356
楼主
发表于 2019-9-23 14:24:10 | 显示全部楼层
这个时间不太正常,建议更新系统固件到V1.5,更新rknn-toolkit到V1.2,再进行测试。
回复

使用道具 举报

troy

版主

积分
2356
沙发
发表于 2019-9-23 17:31:37 | 显示全部楼层
kepurSong 发表于 2019-9-23 14:41
为什么我在PC电脑上用pre_compile=True加速,竟然报错了?说是不能再模拟器上面加速环境是,rknn1.1.0, Ub ...

文档里面有写,开启预编译,就不能在PC上模拟了。运行提示也写的很清楚了,PC模拟器上不能运行预编译的模型。
回复

使用道具 举报

troy

版主

积分
2356
板凳
发表于 2019-9-24 08:35:11 | 显示全部楼层
kepurSong 发表于 2019-9-23 20:47
你理解错了我的问题,我问的是:
1,以前我没有预编译,直接在rk3399pro上面运行,发现每次模型的init_ru ...

我看懂你的意思了。
1.pre_compile是可以加快init_runtime的速度,预编译报的错误是TensorFlow相关的,你可以看下最后几行的日志,看起来是跟cuDNN环境有关系,确认一下TensorFlow的执行环境是否正常。因为在开启量化和预编译等操作时,rknn会调用TensorFlow跑一遍推理,显然此处TensorFlow报错了。
2.预编译是可以在PC上没错。你也可以尝试一下使用最新最新发布rknn-toolkit1.2.0在PC上预编译试一下。但是建议先排除TensorFlow的问题。
回复

使用道具 举报

troy

版主

积分
2356
地板
发表于 2019-9-24 10:22:02 | 显示全部楼层
kepurSong 发表于 2019-9-24 10:19
你这样一说,我好像明白为什么报错了:
1,我Ubuntu上面只有tensorflow-gpu==1.13.1的版本,没有cpu的版 ...

1.你可以自己找个TensorFlow的demo,跑一下TensorFlow,确保你的TensorFlow环境正常
2.rknn-toolkit1.1.0我有用过1.12和1.14版本的TensorFlow都是可以的,我觉得首先还是要把TensorFlow环境弄好,去百度一下这些cuDNN的错误,先解决一下。
回复

使用道具 举报

troy

版主

积分
2356
5#
发表于 2019-9-24 16:28:54 | 显示全部楼层
kepurSong 发表于 2019-9-24 15:26
感谢大神的指导,我修改了一个tensorflow的版本,发现可以预编译成功了!
令人惊讶的是,我YOLOv3的70Mrk ...

NPU最终执行的其实都是预编译后的代码,只是如果rknn模型没开预编译,就会导致在init_runtime的时候执行预编译过程,这个过程在板子上执行确实会比较慢,但是像你这么慢要700s的还是第一次见
回复

使用道具 举报

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

本版积分规则

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


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