我先说一下背景: 1.前阵子我把 rk3399 的相关环境装好了( opencv,tensorflow )。
2.而后震惊的发现板子 arm64 不支持直接编译安装 tensorflow-gpu。
3.后台搜寻了一下,发现 arm64 可以把模型文件生成一个 npu 文件,以供调用
4.但是我只看到了调用 npu api 的 cpp 版本,但是我主工程项目是 python 的,请问如何试好?
最终需求:在 rk3399 上使用加速器( gpu/npu )进行网络推理,加速项目性能, [主工程项目太过于复杂,是 py 的,时间关系没法用 cpp 重写]
1
xiexingjia 2020-02-20 01:50:30 +08:00
是 rk3399pro 吗,rknn 有 Python 绑定的。
|
2
gzchen OP @xiexingjia rk3399 呢,我看了下,rknn toolkit 支持,我尝试下哈~
|
3
gzchen OP @xiexingjia 你用过吗,方便加个微信后续交流?
|
4
rocbomb 2020-02-20 03:40:20 +08:00
这种东西哎 坑太多了
|