Función PyTorch torch.cuda.empty_cache

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

torch.cuda.empty_cacheEs una función de PyTorch que se utiliza para vaciar la caché de CUDA. Libera la memoria caché no utilizada para que pueda reasignarse.

Definición de la función

torch.cuda.empty_cache()

Ejemplo de uso

Ejemplo

import torch

# Vaciar la caché de CUDA
if torch.cuda.is_available():
    # Crear algunos tensores
    x = torch.randn(1000, 1000).cuda()
    y = torch.randn(1000, 1000).cuda()

    # Vaciar la caché
    torch.cuda.empty_cache()
    print("CUDA cache cleared")

    # Comprobar la memoria
    print(f"Memory allocated: {torch.cuda.memory_allocated() / 1024**2:.2f} MB")
else:
    print("CUDA not available")

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

Otras extensiones