rk3399ubuntu18.04 pytorch环境搭建

-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一直解决不了，有大佬知道该怎么弄吗

[人工智能] rk3399ubuntu18.04 pytorch环境搭建