torch.accelerator.set_device_index

torch.accelerator.set_device_index(device, /)

将当前设备索引设置为给定设备。

参数

device (torch.device, str, int) – 给定设备，必须与当前 加速器 设备类型匹配。

注意

如果此设备索引为负数，则此函数不执行任何操作。