Fonction torch.cuda.synchronize de PyTorch
Manuel de référence Pytorch torch
torch.cuda.synchronizeest une fonction de PyTorch utilisée pour la synchronisation CUDA. Elle attend que tous les cœurs CUDA terminent la tâche en cours, garantissant que les opérations CPU et GPU sont synchronisées.
Définition de fonction
torch.cuda.synchronize(device=None)
Exemple d'utilisation
Exemple
import torch
import time
# Exemple de synchronisation CUDA
if torch.cuda.is_available():
# Créer un tenseur CUDA et effectuer des opérations
x = torch.randn(1000, 1000).cuda()
y = torch.mm(x, x)
# Synchroniser les opérations CUDA
torch.cuda.synchronize()
print("Opérations CUDA synchronisées")
else:
print("CUDA non disponible")
import time
# Exemple de synchronisation CUDA
if torch.cuda.is_available():
# Créer un tenseur CUDA et effectuer des opérations
x = torch.randn(1000, 1000).cuda()
y = torch.mm(x, x)
# Synchroniser les opérations CUDA
torch.cuda.synchronize()
print("Opérations CUDA synchronisées")
else:
print("CUDA non disponible")
Autres extensions