Функция torch.cuda.empty_cache в PyTorch

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

torch.cuda.empty_cacheЭто функция в PyTorch, используемая для очистки кэша CUDA. Она освобождает неиспользуемую кэшированную память, чтобы её можно было перераспределить.

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

torch.cuda.empty_cache()

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

Пример

import torch

# Очистить кэш CUDA
if torch.cuda.is_available():
    # Создать несколько тензоров
    x = torch.randn(1000, 1000).cuda()
    y = torch.randn(1000, 1000).cuda()

    # Очистить кэш
    torch.cuda.empty_cache()
    print("Кэш CUDA очищен")

    # Проверить память
    print(f"Выделено памяти: {torch.cuda.memory_allocated() / 1024**2:.2f} MB")
else:
    print("CUDA недоступна")

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

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