Función PyTorch torch.cuda.manual_seed

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.cuda.manual_seedEs la función en PyTorch para establecer la semilla aleatoria de CUDA. Establece la semilla aleatoria para el dispositivo CUDA actual, asegurando la reproducibilidad de las operaciones CUDA.

Definición de la función

torch.cuda.manual_seed(seed)

Ejemplo de uso

Ejemplo

import torch

# Establecer semilla aleatoria de CUDA
if torch.cuda.is_available():
    torch.cuda.manual_seed(42)
    x = torch.randn(3, 3).cuda()
    print(f"Random tensor on CUDA: {x}")

    # La misma semilla produce el mismo resultado
    torch.cuda.manual_seed(42)
    y = torch.randn(3, 3).cuda()
    print(f"Random tensor with same seed: {y}")
else:
    print("CUDA not available")

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones