PyTorch torch.cuda.memory_reserved función

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.cuda.memory_reservedEs una función en PyTorch que se utiliza para obtener la memoria de video reservada. Devuelve el tamaño de la memoria actualmente reservada por el administrador de caché de CUDA (en bytes).

Definición de la función

torch.cuda.memory_reserved(device=None)

Ejemplo de uso

Ejemplo

import torch

# Comprobar la memoria de video reservada
if torch.cuda.is_available():
    print(f"Initial memory reserved: {torch.cuda.memory_reserved()} bytes")

    # Crear tensor
    x = torch.randn(1000, 1000).cuda()
    print(f"After allocation: {torch.cuda.memory_reserved()} bytes")

    # Eliminar tensor
    del x
    print(f"After deletion: {torch.cuda.memory_reserved()} bytes")
else:
    print("CUDA not available")

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones