【随机数种子】random.seed() torch.manual_seed() torch.cuda.manual_seed() torch.cuda.manual_seed_all()用法总结
四种作用相似,先做下概括:
- 传入的数值用于指定随机数生成时所用算法开始时所选定的整数值,如果使用相同的seed()值,则每次生成的随机数都相同;
- 如果不设置这个值,则系统会根据时间来自己选择这个值,此时每次生成的随机数会因时间的差异而有所不同。
random.seed() | random模块的随机数种子 |
torch.manual_seed() | 为CPU设置随机数种子 |
torch.cuda.manual_seed() | 为GPU设置随机数种子 |
torch.cuda.manual_seed_all() | 为所有的GPU设置随机数种子 |