Toybrick»开源社区 › 技术论坛 › 人工智能 › 怎么安装pytorch

怎么安装pytorch

积分: 74

楼主

发表于 2020-11-12 11:52:21 查看: 8859|回复: 1 | [复制链接]

| 只看该作者

rk3399pro可以用pytorch模型了吗，怎么安装pytorch？其他框架比如caffee怎么装，还是说只能进行权重转换，转成rknn才能在板子上运行，大佬们有知道的吗

回复

使用道具举报

积分: 13580

沙发

发表于 2020-11-12 12:14:00 | 只看该作者

需要转成rknn才能运行在我们NPU上。
安装pytorch、tensorflow都没问题，跑的是CPU而已

回复

使用道具举报

返回列表

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

瑞芯微电子股份有限公司所有闽ICP备19006074号-1 闽公网安备 35010202001061号

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