谢谢楼主,这个问题已经解决了,请问这个roc-rk3399-pc在跑深度学习的模型的时候有没有加速电路?pytorch ...
你是说调用gpu或是npu(用扩展版接上的那个?),用pytorch是不能了,现在瑞芯微官方主力推3399pro,集成了npu,他们用自有框架rknn可以调用npu。装了pytorch玩玩得了,加速不用想了 楼主您好!想请问您,我是在Pytorch下面自己修改的网络结构,我看官网说要转换成RKNN Toolkit模型才可以在npu上运行。保存好的预训练模型是.pkl格式的,可以转换吗?转换后在Python中直接调用就可以吗? 我是ubunt20的系统然后python版本为3.8,opencv按照成功了但是pytorch一直解决不了,有大佬知道该怎么弄吗
页:
1
[2]