Toybrick

标题: op自定义层的在linux64位生成的模型，无法在rk3399proD上跑起来 [打印本页]

作者: warrenwg 时间: 2019-11-8 18:40
标题: op自定义层的在linux64位生成的模型，无法在rk3399proD上跑起来

现象：在rknn_toolkit1.2.1下，使用rknn_custom_op_math样例，可以在ubuntu64上编译并生成对应的rknn；但是将这个rknn放到rk3399pro上，使用rknn_toolkit 1.2.1，模型无法加载。

版本信息:

固件TB-RK3399ProD-v1.5

rknn_toolkit1.2.1

D RKNNAPI: RKNN VERSION:

D RKNNAPI: API: 1.2.0 (db63ace build: 2019-09-17 14:23:41)

D RKNNAPI: DRV: 0.9.9 (8509ec7 build: 2019-08-05 10:54:34)

作者: jefferyzhang 时间: 2019-11-8 23:30
1. ubuntu 上用的rknn toolkit是什么版本？
2. 然后你这里报的错是没找到设备，要么是你代码写的有问题，target写错了，要么就是npu没识别到

作者: warrenwg 时间: 2019-11-11 18:13

jefferyzhang 发表于 2019-11-8 23:30
1. ubuntu 上用的rknn toolkit是什么版本？
2. 然后你这里报的错是没找到设备，要么是你代码写的有问题，ta ...

rknn toolkit 是1.2.1,代码是直接用的rknn_custom_op_math的样例代码，我没做任何修改的。另外npu能正常工作，只是在跑自定义op这个demo的时候不行

作者: warrenwg 时间: 2019-11-11 18:17

jefferyzhang 发表于 2019-11-8 23:30
1. ubuntu 上用的rknn toolkit是什么版本？
2. 然后你这里报的错是没找到设备，要么是你代码写的有问题，ta ...

或者请问你们有没有整套能把op的demo跑起来的环境列表，我按照需求一个个去搭建

作者: jefferyzhang 时间: 2019-11-11 20:03
你可以把你代码全部打包下发上来，我发给npu部门看下。
你环境是没有问题的。

作者: warrenwg 时间: 2019-11-12 11:13

jefferyzhang 发表于 2019-11-11 20:03
你可以把你代码全部打包下发上来，我发给npu部门看下。
你环境是没有问题的。 ...

代码附上，是你们op的demo代码，在ubuntu64上可以生成rknnop，并且可以运行，但是在板子上的linux系统上就是init失败（-9）

作者: jefferyzhang 时间: 2019-11-12 11:36
已上报

欢迎光临 Toybrick (https://t.rock-chips.com/)