【随机数种子】random.seed() torch.manual_seed() torch.cuda.manual_seed() torch.cuda.manual_seed_all()用法总结

【随机数种子】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设置随机数种子

你可能感兴趣的:(python,深度学习,算法)