想问下基于 rk3399( Linux )生成的 npu 文件,可以被 Python 调用起来吗?

2020-02-20 01:07:05 +08:00
 gzchen

我先说一下背景: 1.前阵子我把 rk3399 的相关环境装好了( opencv,tensorflow )。

2.而后震惊的发现板子 arm64 不支持直接编译安装 tensorflow-gpu。

3.后台搜寻了一下,发现 arm64 可以把模型文件生成一个 npu 文件,以供调用

4.但是我只看到了调用 npu api 的 cpp 版本,但是我主工程项目是 python 的,请问如何试好?

最终需求:在 rk3399 上使用加速器( gpu/npu )进行网络推理,加速项目性能, [主工程项目太过于复杂,是 py 的,时间关系没法用 cpp 重写]

2177 次点击
所在节点    程序员
4 条回复
xiexingjia
2020-02-20 01:50:30 +08:00
是 rk3399pro 吗,rknn 有 Python 绑定的。
gzchen
2020-02-20 02:22:51 +08:00
@xiexingjia rk3399 呢,我看了下,rknn toolkit 支持,我尝试下哈~
gzchen
2020-02-20 02:25:20 +08:00
@xiexingjia 你用过吗,方便加个微信后续交流?
rocbomb
2020-02-20 03:40:20 +08:00
这种东西哎 坑太多了

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/645980

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX