Toybrick

怎么安装pytorch

Dream_BJ

注册会员

积分
74
发表于 2020-11-12 11:52:21    查看: 5778|回复: 1 | [复制链接]    打印 | 显示全部楼层
rk3399pro可以用pytorch模型了吗,怎么安装pytorch?其他框架比如caffee怎么装,还是说只能进行权重转换,转成rknn才能在板子上运行,大佬们有知道的吗
回复

使用道具 举报

jefferyzhang

版主

积分
12844
发表于 2020-11-12 12:14:00 | 显示全部楼层
需要转成rknn才能运行在我们NPU上。
安装pytorch、tensorflow都没问题,跑的是CPU而已
回复

使用道具 举报

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

本版积分规则

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


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