gpt4 book ai didi

python - torch.manual_seed 是否包含 torch.cuda.manual_seed_all 的操作?

转载 作者:行者123 更新时间:2023-12-04 03:26:09 30 4
gpt4 key购买 nike

是否torch.manual_seed包括torch.cuda.manual_seed_all的操作?
如果是,我们可以使用 torch.manual_seed设置种子。否则我们应该调用这两个函数。

最佳答案

是的, 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
  • 关于python - torch.manual_seed 是否包含 torch.cuda.manual_seed_all 的操作?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/67581281/

    30 4 0
    Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
    广告合作:1813099741@qq.com 6ren.com