Наличие видеокарты в PyTorch

Перенос вычислений на видеокарту имеет смысл только когда драйвер и железо доступны среде. Функция is_available из подмодуля cuda возвращает логический ответ на этот вопрос.

Выведем результат проверки:

import torch

print(torch.cuda.is_available())
# на этой машине False

Если ответ ложный, дальше в примерах раздела остаются вычисления на процессоре. Проверку всё равно стоит ставить в начале скрипта.

Выведите, видит ли PyTorch видеокарту в текущей среде.

Сохраните ответ проверки в переменную и выведите её тип.

Дважды подряд выведите результат той же проверки, каждое значение на своей строке.