PyTorch torch.cuda.manual_seed_all Funktion
PyTorch torch Referenzhandbuch
torch.cuda.manual_seed_allEs ist eine Funktion in PyTorch, mit der der Zufallsseed für alle CUDA-Geräte festgelegt wird. Legen Sie den Zufallsseed für alle CUDA-Geräte fest.
Funktionsdefinition
torch.cuda.manual_seed_all(seed)
Anwendungsbeispiel
Beispiel
import torch
# Zufallsseeds für alle CUDA-Geräte setzen
if torch.cuda.is_available():
torch.cuda.manual_seed_all(42)
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")
# Zufallsseeds für alle CUDA-Geräte setzen
if torch.cuda.is_available():
torch.cuda.manual_seed_all(42)
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