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

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

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

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

torch.cuda.memory_allocated(device=None)

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

Пример

import torch

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

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

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

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

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