Pytorch程序迁移
收藏回复举报
Pytorch程序迁移
t('forum.solved') 已解决
发表于2023-08-21 13:58:57
0 查看

目前我是正在学习Atlas 200I DK A2开发者套件的使用,想把Pytorch框架写的程序迁移到开发板上,直接放在开发板上跑的话太慢了,后来猜想到了可能是因为没有调用npu,所以想了解以下开发者套件芯片中的NPU该如何调用?

看之前的论坛中有人询问torch_npu的问题,发现310B4没有适配。那开发板上想要调用npu的话该怎么办?有没有其他的办法?

我要发帖子