Функция torch.cuda.is_available в PyTorch
Справочное руководство Pytorch torch
torch.cuda.is_availableявляется функцией в PyTorch, используемой для проверки доступности CUDA.
Определение функции
torch.cuda.is_available()
Пример использования
Пример
import torch
# Проверить, доступна ли CUDA
if torch.cuda.is_available():
print("CUDA доступна!")
print("Количество GPU:", torch.cuda.device_count())
print("Имя GPU:", torch.cuda.get_device_name(0))
else:
print("CUDA недоступна")
# Проверить, доступна ли CUDA
if torch.cuda.is_available():
print("CUDA доступна!")
print("Количество GPU:", torch.cuda.device_count())
print("Имя GPU:", torch.cuda.get_device_name(0))
else:
print("CUDA недоступна")
Другие расширения