华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
我将N卡(Pytorch)的模型迁移过来,发现某些简单的模型训练速度变快了,但某个复杂点的模型训练速度慢很多。 使用MindStudio Insight查看如下图 发现NPU一直在等待,好像很多算子是CPU在跑?有些名字大概能出来对应什么模块,但哪里能查到对应。
我要发帖子
我将N卡(Pytorch)的模型迁移过来,发现某些简单的模型训练速度变快了,但某个复杂点的模型训练速度慢很多。

使用MindStudio Insight查看如下图
发现NPU一直在等待,好像很多算子是CPU在跑?有些名字大概能出来对应什么模块,但哪里能查到对应。