Функция PyTorch torch.cuda.memory_reserved

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

torch.cuda.memory_reservedЭто функция в PyTorch, используемая для получения зарезервированной памяти GPU. Она возвращает размер памяти, в настоящее время зарезервированной менеджером кэширования CUDA (в байтах).

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

torch.cuda.memory_reserved(device=None)

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

Пример

import torch

# Проверить зарезервированную память GPU
if torch.cuda.is_available():
    print(f"Начальная зарезервированная память: {torch.cuda.memory_reserved()} bytes")

    # Создать тензор
    x = torch.randn(1000, 1000).cuda()
    print(f"После выделения: {torch.cuda.memory_reserved()} bytes")

    # Удалить тензор
    del x
    print(f"После удаления: {torch.cuda.memory_reserved()} bytes")
else:
    print("CUDA недоступна")

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

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