Función PyTorch torch.cuda.manual_seed_all
Manual de referencia de Pytorch torch
torch.cuda.manual_seed_allEs una función en PyTorch que se utiliza para establecer la semilla aleatoria de todos los dispositivos CUDA. Establece la semilla aleatoria para todos los dispositivos CUDA.
Definición de la función
torch.cuda.manual_seed_all(seed)
Ejemplo de uso
Ejemplo
import torch
# Establece la semilla aleatoria para todos los dispositivos CUDA
if torch.cuda.is_available():
torch.cuda.manual_seed_all(42)
x = torch.randn(3, 3).cuda()
print(f"Random tensor: {x}")
print(f"CUDA device count: {torch.cuda.device_count()}")
else:
print("CUDA not available")
# Establece la semilla aleatoria para todos los dispositivos CUDA
if torch.cuda.is_available():
torch.cuda.manual_seed_all(42)
x = torch.randn(3, 3).cuda()
print(f"Random tensor: {x}")
print(f"CUDA device count: {torch.cuda.device_count()}")
else:
print("CUDA not available")
Otras extensiones