Toybrick

标题: 怎么安装pytorch [打印本页]

作者: Dream_BJ 时间: 2020-11-12 11:52
标题: 怎么安装pytorch
rk3399pro可以用pytorch模型了吗，怎么安装pytorch？其他框架比如caffee怎么装，还是说只能进行权重转换，转成rknn才能在板子上运行，大佬们有知道的吗

作者: jefferyzhang 时间: 2020-11-12 12:14
需要转成rknn才能运行在我们NPU上。
安装pytorch、tensorflow都没问题，跑的是CPU而已

欢迎光临 Toybrick (https://t.rock-chips.com/)