torch.accelerator.set_device_index

torch.accelerator.set_device_index(device, /)

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

参数

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

注意

如果此设备索引为负，则此函数为空操作。