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)

讓我相信這些都是而且只是所需的種子。