torch.accelerator.current_device_idx

torch.accelerator.current_device_idx() [исходный код]

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

Возвращает:

индекс текущего выбранного устройства.

Тип возвращаемого значения:

int

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

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