Fonction torch.cuda.manual_seed de PyTorch
Manuel de référence Pytorch torch
torch.cuda.manual_seedest une fonction dans PyTorch utilisée pour définir la graine aléatoire CUDA. Elle définit la graine aléatoire pour l’appareil CUDA actuel afin de garantir la reproductibilité des opérations CUDA.
Définition de fonction
torch.cuda.manual_seed(seed)
Exemple d'utilisation
Exemple
import torch
# Définir la graine aléatoire CUDA
if torch.cuda.is_available():
torch.cuda.manual_seed(42)
x = torch.randn(3, 3).cuda()
print(f"Tenseur aléatoire sur CUDA : {x}")
# La même graine produit le même résultat
torch.cuda.manual_seed(42)
y = torch.randn(3, 3).cuda()
print(f"Tenseur aléatoire avec la même graine : {y}")
else:
print("CUDA non disponible")
# Définir la graine aléatoire CUDA
if torch.cuda.is_available():
torch.cuda.manual_seed(42)
x = torch.randn(3, 3).cuda()
print(f"Tenseur aléatoire sur CUDA : {x}")
# La même graine produit le même résultat
torch.cuda.manual_seed(42)
y = torch.randn(3, 3).cuda()
print(f"Tenseur aléatoire avec la même graine : {y}")
else:
print("CUDA non disponible")
Autres extensions