PyTorch torch.cuda.memory_allocated Funktion
PyTorch torch Referenzhandbuch
torch.cuda.memory_allocatedist eine Funktion in PyTorch, mit der der zugewiesene GPU-Speicher abgerufen wird. Sie gibt die Größe des auf dem aktuellen CUDA-Gerät zugewiesenen Speichers zurück (in bytes).
Funktionsdefinition
torch.cuda.memory_allocated(device=None)
Anwendungsbeispiel
Beispiel
import torch
# Zugewiesenen GPU-Speicher prüfen
if torch.cuda.is_available():
print(f"Initialer Speicher: {torch.cuda.memory_allocated()} bytes")
# Tensor erstellen
x = torch.randn(1000, 1000).cuda()
print(f"Nach der Zuweisung: {torch.cuda.memory_allocated()} bytes")
# Tensor löschen
del x
print(f"Nach dem Löschen: {torch.cuda.memory_allocated()} bytes")
else:
print("CUDA nicht verfügbar")
# Zugewiesenen GPU-Speicher prüfen
if torch.cuda.is_available():
print(f"Initialer Speicher: {torch.cuda.memory_allocated()} bytes")
# Tensor erstellen
x = torch.randn(1000, 1000).cuda()
print(f"Nach der Zuweisung: {torch.cuda.memory_allocated()} bytes")
# Tensor löschen
del x
print(f"Nach dem Löschen: {torch.cuda.memory_allocated()} bytes")
else:
print("CUDA nicht verfügbar")
Andere Erweiterungen