torch.accelerator.set_device_idx

torch.accelerator.set_device_idx(device, /) [source]

(Устарело) Устанавливает текущий индекс устройства для заданного устройства.

Параметры:

device (torch.device, str, int) – заданное устройство, тип которого должен соответствовать типу текущего устройства ускорителя.

Предупреждение

torch.accelerator.set_device_idx() устарел; вместо него следует использовать torch.accelerator.set_device_index(). В одном из будущих выпусков PyTorch он будет удалён.