华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
目前我是正在学习Atlas 200I DK A2开发者套件的使用,想把Pytorch框架写的程序迁移到开发板上,直接放在开发板上跑的话太慢了,想了解以下开发者套件芯片中的NPU该如何调用?
看之前的论坛中有人询问torch_npu的问题,发现310B4没有适配。那开发板上想要调用npu的话该怎么办?
本帖最后由 匿名用户 于 2023/08/19 17:17:10 编辑
我要发帖子
目前我是正在学习Atlas 200I DK A2开发者套件的使用,想把Pytorch框架写的程序迁移到开发板上,直接放在开发板上跑的话太慢了,想了解以下开发者套件芯片中的NPU该如何调用?
看之前的论坛中有人询问torch_npu的问题,发现310B4没有适配。那开发板上想要调用npu的话该怎么办?