PyTorch torch.device Funktion
PyTorch torch-Referenzhandbuch
torch.deviceist eine Funktion in PyTorch, die zum Erstellen von Geräteobjekten verwendet wird und dazu dient, das Gerät anzugeben, auf dem Tensoren gespeichert werden.
Funktionsdefinition
torch.device(device)
Anwendungsbeispiele
Beispiele
import torch
# CPU-Gerät erstellen
cpu_device = torch.device('cpu')
print("CPU-Gerät:", cpu_device)
# CUDA-Gerät erstellen
if torch.cuda.is_available():
cuda_device = torch.device('cuda:0')
print("CUDA-Gerät:", cuda_device)
# Tensor auf GPU erstellen
x = torch.zeros(3, 4, device=cuda_device)
print("Gerät des Tensors:", x.device)
# CPU-Gerät erstellen
cpu_device = torch.device('cpu')
print("CPU-Gerät:", cpu_device)
# CUDA-Gerät erstellen
if torch.cuda.is_available():
cuda_device = torch.device('cuda:0')
print("CUDA-Gerät:", cuda_device)
# Tensor auf GPU erstellen
x = torch.zeros(3, 4, device=cuda_device)
print("Gerät des Tensors:", x.device)
Andere Erweiterungen