PyTorch torch.cuda.manual_seed-Funktion
PyTorch torch Referenzhandbuch
torch.cuda.manual_seedist eine Funktion in PyTorch, mit der der CUDA-Zufallsseed festgelegt wird. Sie legt den Zufallsseed für das aktuelle CUDA-Gerät fest, um die Reproduzierbarkeit von CUDA-Operationen zu gewährleisten.
Funktionsdefinition
torch.cuda.manual_seed(seed)
Anwendungsbeispiel
Beispiel
import torch
# CUDA-Zufallsseed festlegen
if torch.cuda.is_available():
torch.cuda.manual_seed(42)
x = torch.randn(3, 3).cuda()
print(f"Zufälliger Tensor auf CUDA: {x}")
# Derselbe Seed erzeugt dasselbe Ergebnis
torch.cuda.manual_seed(42)
y = torch.randn(3, 3).cuda()
print(f"Zufälliger Tensor mit demselben Seed: {y}")
else:
print("CUDA nicht verfügbar")
# CUDA-Zufallsseed festlegen
if torch.cuda.is_available():
torch.cuda.manual_seed(42)
x = torch.randn(3, 3).cuda()
print(f"Zufälliger Tensor auf CUDA: {x}")
# Derselbe Seed erzeugt dasselbe Ergebnis
torch.cuda.manual_seed(42)
y = torch.randn(3, 3).cuda()
print(f"Zufälliger Tensor mit demselben Seed: {y}")
else:
print("CUDA nicht verfügbar")
Andere Erweiterungen