PyTorch torch.cuda-Funktionen
PyTorch torch Referenzhandbuch
torch.cudaEs ist ein Modul in PyTorch für die CUDA-Verwaltung. Es bietet Funktionen wie CUDA-Geräteverwaltung, Speicherverwaltung und Synchronisierung.
Funktionsdefinition
torch.cuda
Das Modul torch.cuda enthält die folgenden häufig verwendeten Funktionen:
torch.cuda.is_available()- Prüfen, ob CUDA verfügbar isttorch.cuda.device_count()- Anzahl der CUDA-Geräte abrufentorch.cuda.current_device()- Aktuelles Gerät abrufentorch.cuda.synchronize()- CUDA-Operationen synchronisierentorch.cuda.empty_cache()- Cache leeren
Anwendungsbeispiel
Beispiel
import torch
# Prüfen, ob CUDA verfügbar ist
print(f"CUDA verfügbar: {torch.cuda.is_available()}")
# Anzahl der CUDA-Geräte abrufen
if torch.cuda.is_available():
print(f"Anzahl der CUDA-Geräte: {torch.cuda.device_count()}")
print(f"CUDA-Gerätename: {torch.cuda.get_device_name(0)}")
# Prüfen, ob CUDA verfügbar ist
print(f"CUDA verfügbar: {torch.cuda.is_available()}")
# Anzahl der CUDA-Geräte abrufen
if torch.cuda.is_available():
print(f"Anzahl der CUDA-Geräte: {torch.cuda.device_count()}")
print(f"CUDA-Gerätename: {torch.cuda.get_device_name(0)}")
Andere Erweiterungen