Функция torch.is_available в PyTorch

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.is_available— это функция в PyTorch, используемая для проверки доступности CUDA. Возвращает True, если CUDA доступна, и False в противном случае.

Определение функции

torch.is_available()

Пример использования

Пример

import torch

# Проверить, доступна ли CUDA
if torch.is_available():
    print("CUDA is available!")
    print(f"Количество устройств CUDA: {torch.cuda.device_count()}")
else:
    print("CUDA is not available")

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения