torch.cuda.manual_seed

torch.cuda.manual_seed(seed)[源代码]

为当前GPU生成随机数设置种子。

如果 CUDA 不可用，调用此函数是安全的；在这种情况下，该函数会被默默忽略。

参数

seed (int) – 指定的随机种子值。

警告

如果你正在处理一个多GPU模型，仅使用当前函数无法保证确定性。为了为所有GPU设置种子，请使用manual_seed_all()。