torch.accelerator.set_device_index#

torch.accelerator.set_device_index(device,/)[source]#

Set the current device index to a given device.

Parameters

device (torch.device, str, int) – a given device that must match the currentaccelerator device type.

Note

This function is a no-op if this device index is negative.