PyTorch torch.cuda.empty_cache Funktion
PyTorch torch-Referenzhandbuch
torch.cuda.empty_cacheEs ist eine Funktion in PyTorch, die verwendet wird, um den CUDA-Cache zu leeren. Sie gibt ungenutzten Cache-Speicher frei, damit er neu zugewiesen werden kann.
Funktionsdefinition
torch.cuda.empty_cache()
Anwendungsbeispiel
Beispiel
import torch
# CUDA-Cache leeren
if torch.cuda.is_available():
# Einige Tensoren erstellen
x = torch.randn(1000, 1000).cuda()
y = torch.randn(1000, 1000).cuda()
# Den Cache leeren
torch.cuda.empty_cache()
print("CUDA-Cache geleert")
# Speicher überprüfen
print(f"Zugewiesener Speicher: {torch.cuda.memory_allocated() / 1024**2:.2f} MB")
else:
print("CUDA nicht verfügbar")
# CUDA-Cache leeren
if torch.cuda.is_available():
# Einige Tensoren erstellen
x = torch.randn(1000, 1000).cuda()
y = torch.randn(1000, 1000).cuda()
# Den Cache leeren
torch.cuda.empty_cache()
print("CUDA-Cache geleert")
# Speicher überprüfen
print(f"Zugewiesener Speicher: {torch.cuda.memory_allocated() / 1024**2:.2f} MB")
else:
print("CUDA nicht verfügbar")
Andere Erweiterungen