×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

怎么使用多个gpu并行计算 z5925178302024-08-26 13:51:56 回复 1 查看 使用求助
怎么使用多个gpu并行计算
分享到:

现配置了多个gpu,统一型号,cnmon可以查看到,但我每次跑程序都只用了一个gpu,怎么同时使用多个gpu进行并行计算?


现在主要代码

import torch

import torch_mlu

...

data = data.to(torch.device("mlu"))

model = model.to(torch.device("mlu"))




版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭