torch.accelerator.device_count

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

Возвращает количество доступных текущих ускорителей.

Возвращаемое значение:

количество доступных текущих ускорителей.

Если доступных ускорителей нет, возвращает 0.

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

int

Примечание

Этот API делегирует вызов версии device_count, специфичной для устройства. В CUDA этот API НЕ нарушит работу fork, если обнаружение NVML завершится успешно. В противном случае нарушит. Подробнее см. в разделе Нарушение работы fork при многопроцессорной обработке.