Fonction torch.cuda.manual_seed_all de PyTorch
Manuel de référence Pytorch torch
torch.cuda.manual_seed_allIl s'agit d'une fonction de PyTorch utilisée pour définir la graine aléatoire pour tous les périphériques CUDA. Définir la graine aléatoire pour tous les périphériques CUDA.
Définition de fonction
torch.cuda.manual_seed_all(seed)
Exemple d'utilisation
Exemple
import torch
# Définir les graines aléatoires pour tous les périphériques CUDA
if torch.cuda.is_available():
torch.cuda.manual_seed_all(42)
x = torch.randn(3, 3).cuda()
print(f"Tenseur aléatoire : {x}")
print(f"Nombre de périphériques CUDA : {torch.cuda.device_count()}")
else:
print("CUDA non disponible")
# Définir les graines aléatoires pour tous les périphériques CUDA
if torch.cuda.is_available():
torch.cuda.manual_seed_all(42)
x = torch.randn(3, 3).cuda()
print(f"Tenseur aléatoire : {x}")
print(f"Nombre de périphériques CUDA : {torch.cuda.device_count()}")
else:
print("CUDA non disponible")
Autres extensions