torch.manual_seed 是否包含 torch.cuda.manual_seed_all 的操作？

Question

torch.manual_seed是否包含torch.cuda.manual_seed_all的操作？

如果是，我们可以使用torch.manual_seed来设置种子。 否则我们应该调用这两个函数。

Answer 1

是的， torch.manual_seed()确实包括 CUDA：

您可以使用torch.manual_seed()为所有设备（CPU 和 CUDA）播种 RNG：

• https://pytorch.org/docs/stable/notes/randomness.html

Answer 2

请参阅Pytorch 闪电的seed_everything ：

random.seed(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed_all(seed)

让我相信这些都是而且只是所需的种子。