torch.accelerator.set_device_index

torch.accelerator.set_device_index(device, /) [исходный код]

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

Параметры:

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

Примечание

Эта функция ничего не делает, если индекс устройства отрицательный.