torch.accelerator.set_device_idx

torch.accelerator.set_device_idx(device, /)

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

参数

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

注意

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