-Forest- 发表于 2020-4-23 21:25:05

Wanggp 发表于 2020-4-22 22:06
谢谢楼主,这个问题已经解决了,请问这个roc-rk3399-pc在跑深度学习的模型的时候有没有加速电路?pytorch ...

你是说调用gpu或是npu(用扩展版接上的那个?),用pytorch是不能了,现在瑞芯微官方主力推3399pro,集成了npu,他们用自有框架rknn可以调用npu。装了pytorch玩玩得了,加速不用想了

SJMa 发表于 2021-7-30 15:37:47

楼主您好!想请问您,我是在Pytorch下面自己修改的网络结构,我看官网说要转换成RKNN Toolkit模型才可以在npu上运行。保存好的预训练模型是.pkl格式的,可以转换吗?转换后在Python中直接调用就可以吗?

zzz1997 发表于 2023-5-9 17:09:08

我是ubunt20的系统然后python版本为3.8,opencv按照成功了但是pytorch一直解决不了,有大佬知道该怎么弄吗
页: 1 [2]
查看完整版本: rk3399ubuntu18.04 pytorch环境搭建