Toybrick

关于pytorch转化RKNN成功,初始化环境出错的问题

18022443868

中级会员

积分
230
楼主
发表于 2020-4-29 16:24:48    查看: 7818|回复: 4 | [复制链接]    打印 | 只看该作者
D:\Anaconda3\envs\rknn\python.exe "D:\PyCharm 2019.1.2\helpers\pydev\pydevd.py" --multiproc --qt-support=auto --client 127.0.0.1 --port 56080 --file G:/googleDownload/RKNN/RKNN/rknn-toolkit/examples/pytorch/resnet18/test.py
pydev debugger: process 12192 is connecting

Connected to pydev debugger (build 191.7141.48)
--> config model
done
--> Loading model
D:\Anaconda3\envs\rknn\lib\site-packages\onnx_tf\common\__init__.py:87: UserWarning: FrontendHandler.get_outputs_names is deprecated. It will be removed in future release.. Use node.outputs instead.
  warnings.warn(message)
./model.pt ********************
WARNING: Token 'COMMENT' defined, but not used
WARNING: There is 1 unused token
done
--> Building model
done
--> Export RKNN model
done
--> Init runtime environment
E RKNNAPI: rknn_init,  msg_load_ack fail, ack = 1(ACK_FAIL), expect 0(ACK_SUCC)!
E RKNNAPI: ==============================================
E RKNNAPI: RKNN VERSION:
E RKNNAPI:   API: 1.3.0 (c5654ea build: 2019-12-25 08:54:18)
E RKNNAPI:   DRV: 0.9.8 (943f8e8 build: 2019-07-03 17:51:17)
E RKNNAPI: ==============================================
E Catch exception when init runtime!
E Traceback (most recent call last):
E   File "rknn\api\rknn_base.py", line 988, in rknn.api.rknn_base.RKNNBase.init_runtime
E   File "rknn\api\rknn_runtime.py", line 320, in rknn.api.rknn_runtime.RKNNRuntime.build_graph
E Exception: RKNN init failed. error code: RKNN_ERR_MODEL_INVALID
E Current device id is: TS018082190800713
E Devices connected:
E ['TS018082190800713']
Init runtime environment failed

进程已结束,退出代码 -1

上述环境:Win10,python3.6 torch1.2.0 rknn-toolkit 1.3.0 ,另外在RK1808计算棒上面就行相关计算。
出现上述错误,这是Bug,还是我模型问题? 理论上,如果模型框架不支持的话,在转化为.rknn文件是会报错误,但是成功运行到初始化函数,这就想不通了。

回复

使用道具 举报

jefferyzhang

版主

积分
13580
沙发
发表于 2020-4-30 08:43:28 | 只看该作者
你这里DRV和API版本不匹配是一定会出问题的,请把DRV版本也升级到1.3.x
回复

使用道具 举报

18022443868

中级会员

积分
230
板凳
 楼主| 发表于 2020-4-30 09:17:42 | 只看该作者
PC端,怎么升级?
回复

使用道具 举报

18022443868

中级会员

积分
230
地板
 楼主| 发表于 2020-4-30 09:39:06 | 只看该作者
本帖最后由 18022443868 于 2020-4-30 10:09 编辑

在RK3399Pro板子上面,DRV和API版本都是1.3.0版本,同样也出现了上述问题。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

18022443868

中级会员

积分
230
5#
 楼主| 发表于 2020-4-30 11:32:19 | 只看该作者
在台式上面这里的DRV是指计算棒的版本吗?Win10 只支持RK1808计算棒
回复

使用道具 举报

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

本版积分规则

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


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