Чтобы проверить, использует ли PyTorch графический процессор, вы можете использовать следующие методы:

Метод 1: использование torch.cuda.is_available()
Вы можете использовать функцию torch.cuda.is_available(), чтобы проверить, доступен ли графический процессор для использования PyTorch. Эта функция возвращает логическое значение, указывающее, доступен ли графический процессор или нет. Вот пример:

import torch
if torch.cuda.is_available():
    print("GPU is available")
else:
    print("GPU is not available")

Метод 2: проверка устройства тензора
Вы также можете проверить устройство, на котором расположен тензор PyTorch. Если тензор находится на графическом процессоре, это означает, что PyTorch использует графический процессор. Вот пример:

import torch
x = torch.tensor([1, 2, 3])
if x.is_cuda:
    print("PyTorch is using GPU")
else:
    print("PyTorch is using CPU")

Метод 3: печать текущего устройства
Вы можете распечатать текущее устройство, используемое PyTorch, с помощью torch.cuda.current_device(). Если эта функция возвращает действительный индекс устройства, это означает, что PyTorch использует графический процессор. Вот пример:

import torch
device = torch.cuda.current_device()
if device != -1:
    print("PyTorch is using GPU")
else:
    print("PyTorch is using CPU")