PyTorch torch.is_available-Funktion
PyTorch torch-Referenzhandbuch
torch.is_availableist eine Funktion in PyTorch, mit der überprüft wird, ob CUDA verfügbar ist. Gibt True zurück, wenn CUDA verfügbar ist, andernfalls False.
Funktionsdefinition
torch.is_available()
Anwendungsbeispiel
Beispiel
import torch
# Prüfen, ob CUDA verfügbar ist
if torch.is_available():
print("CUDA ist verfügbar!")
print(f"Anzahl der CUDA-Geräte: {torch.cuda.device_count()}")
else:
print("CUDA ist nicht verfügbar")
# Prüfen, ob CUDA verfügbar ist
if torch.is_available():
print("CUDA ist verfügbar!")
print(f"Anzahl der CUDA-Geräte: {torch.cuda.device_count()}")
else:
print("CUDA ist nicht verfügbar")
Andere Erweiterungen