ascend910运算速度在cpu和npu上权衡
收藏回复举报
ascend910运算速度在cpu和npu上权衡
t('forum.solved') 已解决
发表于2024-02-02 10:11:20
0 查看

## 背景

在ascend910上设置了mindspore的cpu和npu两个环境,同一个模型(知识图谱模型,数据量大,处理时间较长10-20min),npu版本的mindspore比cpu版的mindspore慢很多(差5,6倍,越到后面越慢直到cpu运算基本跑不了),主要是在前面数据处理的一些纯cpu运算不牵涉反向传播上非常慢。

## 问题

现在认为是npu和cpu之间运行处理的问题,npu版本中是不是程序的一般运算也在npu上运行了,数据传输的过程就把整体拖慢了。

我要发帖子