Fonction PyTorch torch.cuda.memory_reserved

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.cuda.memory_reservedIl s'agit d'une fonction de PyTorch utilisée pour obtenir la mémoire GPU réservée. Elle renvoie la taille de la mémoire actuellement réservée par le gestionnaire de cache CUDA (en bytes).

Définition de fonction

torch.cuda.memory_reserved(device=None)

Exemple d'utilisation

Exemple

import torch

# Vérifier la mémoire GPU réservée
if torch.cuda.is_available():
    print(f"Mémoire initiale réservée : {torch.cuda.memory_reserved()} bytes")

    # Créer un tenseur
    x = torch.randn(1000, 1000).cuda()
    print(f"Après allocation : {torch.cuda.memory_reserved()} bytes")

    # Supprimer le tenseur
    del x
    print(f"Après suppression : {torch.cuda.memory_reserved()} bytes")
else:
    print("CUDA non disponible")

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions