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