python - 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:

You can use torch.manual_seed() to seed the RNG for all devices (both CPU and CUDA):

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