V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
gzchen
V2EX  ›  程序员

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

  •  
  •   gzchen · 2020-02-20 01:07:05 +08:00 · 2355 次点击
    这是一个创建于 1718 天前的主题,其中的信息可能已经有所发展或是发生改变。

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

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

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

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

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

    4 条回复    2020-02-20 03:40:20 +08:00
    xiexingjia
        1
    xiexingjia  
       2020-02-20 01:50:30 +08:00
    是 rk3399pro 吗,rknn 有 Python 绑定的。
    gzchen
        2
    gzchen  
    OP
       2020-02-20 02:22:51 +08:00
    @xiexingjia rk3399 呢,我看了下,rknn toolkit 支持,我尝试下哈~
    gzchen
        3
    gzchen  
    OP
       2020-02-20 02:25:20 +08:00
    @xiexingjia 你用过吗,方便加个微信后续交流?
    rocbomb
        4
    rocbomb  
       2020-02-20 03:40:20 +08:00
    这种东西哎 坑太多了
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   979 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 23ms · UTC 20:24 · PVG 04:24 · LAX 13:24 · JFK 16:24
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.