Чтобы проверить, доступен ли CUDA в PyTorch, вы можете использовать следующие методы:

1. Метод 1: использование torch.cuda.is_available()

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

2. Способ 2. Проверка типа устройства

   import torch
   device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
   print("Device:", device)

3. Метод 3. Получение списка доступных устройств CUDA

   import torch
   if torch.cuda.is_available():
      device_count = torch.cuda.device_count()
      print("Number of CUDA devices:", device_count)
      for i in range(device_count):
          print("CUDA device", i, ":", torch.cuda.get_device_name(i))
   else:
      print("CUDA is not available")

4. Метод 4. Использование torch.version.cudaдля проверки версии CUDA

   import torch
   if torch.cuda.is_available():
      print("CUDA version:", torch.version.cuda)
   else:
      print("CUDA is not available")