PyTorch torch.set_default_device Funktion
PyTorch torch Referenzhandbuch
torch.set_default_deviceEs ist eine Funktion in PyTorch, mit der das Standardgerät festgelegt wird.
Funktionsdefinition
torch.set_default_device(device)
Anwendungsbeispiel
Beispiel
import torch
# Standardgerät festlegen
if torch.cuda.is_available():
torch.set_default_device('cuda')
x = torch.zeros(3, 4)
print("Standardgerät:", x.device)
torch.set_default_device('cpu')
else:
print("CUDA ist nicht verfügbar")
# Standardgerät festlegen
if torch.cuda.is_available():
torch.set_default_device('cuda')
x = torch.zeros(3, 4)
print("Standardgerät:", x.device)
torch.set_default_device('cpu')
else:
print("CUDA ist nicht verfügbar")
Andere Erweiterungen