torch.accelerator.is_available

torch.accelerator.is_available() [source]

Проверяет, доступен ли текущий ускоритель во время выполнения: он был собран, все необходимые драйверы доступны и видно как минимум одно устройство. Подробнее см. в разделе ускоритель.

Возвращает:

Логическое значение, указывающее, доступен ли ускоритель.

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

bool

Примечание

Этот API делегирует вызов версии is_available для конкретного устройства. В CUDA, если задана переменная окружения PYTORCH_NVML_BASED_CUDA_CHECK=1, эта функция НЕ отравляет fork. В противном случае отравляет. Подробнее см. в разделе Отравление fork при использовании multiprocessing.

Пример:

>>> assert torch.accelerator.is_available() "No available accelerators detected."