Функции PyTorch torch.cuda

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

torch.cudaЭто модуль в PyTorch для управления CUDA. Он предоставляет такие функции, как управление устройствами CUDA, управление памятью и синхронизация.

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

torch.cuda

Модуль torch.cuda содержит следующие часто используемые функции:

  • torch.cuda.is_available()- Проверить, доступна ли CUDA
  • torch.cuda.device_count()- Получить количество устройств CUDA
  • torch.cuda.current_device()- Получить текущее устройство
  • torch.cuda.synchronize()- Синхронизировать операции CUDA
  • torch.cuda.empty_cache()- Очистить кэш

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

Пример

import torch

# Проверить, доступна ли CUDA
print(f"CUDA доступна: {torch.cuda.is_available()}")

# Получить количество устройств CUDA
if torch.cuda.is_available():
    print(f"Количество устройств CUDA: {torch.cuda.device_count()}")
    print(f"Имя устройства CUDA: {torch.cuda.get_device_name(0)}")

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

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