PyTorch torch.cuda.is_available-Funktion
PyTorch torch Referenzhandbuch
torch.cuda.is_availableist eine Funktion in PyTorch, mit der überprüft wird, ob CUDA verfügbar ist.
Funktionsdefinition
torch.cuda.is_available()
Anwendungsbeispiel
Beispiel
import torch
# Überprüfen, ob CUDA verfügbar ist
if torch.cuda.is_available():
print("CUDA verfügbar!")
print("Anzahl der GPUs:", torch.cuda.device_count())
print("GPU-Name:", torch.cuda.get_device_name(0))
else:
print("CUDA nicht verfügbar")
# Überprüfen, ob CUDA verfügbar ist
if torch.cuda.is_available():
print("CUDA verfügbar!")
print("Anzahl der GPUs:", torch.cuda.device_count())
print("GPU-Name:", torch.cuda.get_device_name(0))
else:
print("CUDA nicht verfügbar")
Andere Erweiterungen