Función PyTorch torch.cuda.seed
Manual de referencia de Pytorch torch
torch.cuda.seedEs una función en PyTorch que se utiliza para establecer la semilla aleatoria de CUDA. Establece la semilla aleatoria para el dispositivo CUDA actual (la funcionalidad es la misma que manual_seed).
Definición de la función
torch.cuda.seed()
Ejemplo de uso
Ejemplo
import torch
# Establecer la semilla aleatoria de CUDA
if torch.cuda.is_available():
torch.cuda.seed()
x = torch.randn(3, 3).cuda()
print(f"Random tensor on CUDA: {x}")
else:
print("CUDA not available")
# Establecer la semilla aleatoria de CUDA
if torch.cuda.is_available():
torch.cuda.seed()
x = torch.randn(3, 3).cuda()
print(f"Random tensor on CUDA: {x}")
else:
print("CUDA not available")
Otras extensiones