torch.accelerator.set_device_index

torch.accelerator.set_device_index(device, /)

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

参数

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

注意

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