Fonction torch.cuda.seed_all de PyTorch
Manuel de référence Pytorch torch
torch.cuda.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éfinit la graine aléatoire pour tous les périphériques CUDA (fonctionnellement identique à manual_seed_all).
Définition de fonction
torch.cuda.seed_all()
Exemple d'utilisation
Exemple
import torch
# Définir la graine aléatoire pour tous les périphériques CUDA
if torch.cuda.is_available():
torch.cuda.seed_all()
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 la graine aléatoire pour tous les périphériques CUDA
if torch.cuda.is_available():
torch.cuda.seed_all()
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