比较与torch.cuda.set_device的功能差异

查看源文件

torch.cuda.set_device

torch.cuda.set_device(device)

更多内容详见torch.cuda.set_device。

mindspore.set_context

mindspore.set_context(**kwargs)

更多内容详见mindspore.set_context。

使用方式

PyTorch：设置当前使用的device卡号。

MindSpore：不仅设置当前使用的device卡号，还设置模式mode，运行环境device_target，是否保存图save_graphs等。