在 鲲鹏920 + atlas 300t (Ascend 910) 平台上安装 mindspore 后, 如何知道 mindspore 是在基于 cpu 训练 还是 gpu/Ascend 910 训练
收藏回复举报
在 鲲鹏920 + atlas 300t (Ascend 910) 平台上安装 mindspore 后, 如何知道 mindspore 是在基于 cpu 训练 还是 gpu/Ascend 910 训练
t('forum.solved') 已解决
发表于2023-10-16 17:33:06
0 查看

深度学习通常跑在 CPU + GPU, 只要装好了驱动, cuda, cudnn, 装对了版本, tensorflow 可以自动跑在 GPU 上, 不需要代码指定(可以指定, 但不必需)

在 鲲鹏920 + Ascend 910 + mindspore 上, 也可以吗? 还是必须调下面这行代码

ms.set_context(device_target="Ascend")

我客户的公司采购了 鲲鹏920 + atlas 300t 平台, 按文档 https://www.mindspore.cn/install/#guide 安装mindspore

我理解 鲲鹏920 相当于 CPU, Ascend 910 相当于 GPU (如果我理解有误, 请指出), 如何知道 mindspore 是在基于 cpu 训练 还是 gpu 训练?

本帖最后由 匿名用户2023/10/16 17:34:45 编辑

我要发帖子