PyTorch torch.cuda.seed_all Funktion
PyTorch torch Referenzhandbuch
torch.cuda.seed_allEs ist eine Funktion in PyTorch, mit der der Zufallsseed für alle CUDA-Geräte festgelegt wird. Legt den Zufallsseed für alle CUDA-Geräte fest (funktional identisch mit manual_seed_all).
Funktionsdefinition
torch.cuda.seed_all()
Anwendungsbeispiel
Beispiel
import torch
# Den Zufallsseed für alle CUDA-Geräte setzen
if torch.cuda.is_available():
torch.cuda.seed_all()
x = torch.randn(3, 3).cuda()
print(f"Zufälliger Tensor: {x}")
print(f"Anzahl der CUDA-Geräte: {torch.cuda.device_count()}")
else:
print("CUDA nicht verfügbar")
# Den Zufallsseed für alle CUDA-Geräte setzen
if torch.cuda.is_available():
torch.cuda.seed_all()
x = torch.randn(3, 3).cuda()
print(f"Zufälliger Tensor: {x}")
print(f"Anzahl der CUDA-Geräte: {torch.cuda.device_count()}")
else:
print("CUDA nicht verfügbar")
Andere Erweiterungen