pytorch中的torch.manual_seed(myseed) 和 torch.cuda.manual_seed(myseed) 2021-05-10 12:17:34 浏览数 (1) myseed = 45216 使用方法: 为CPU中设置种子,生成随机数 torch.manual_seed(myseed) 为特定GPU设置种子,生成随机数 torch.cuda.manual_seed(myseed) 为所有GPU设置种子,生成随机数 torch.cuda.manual_seed_all(myseed) 解释: 在实验中需要生成随机数据的时候,每次实验都需要生成数据。设置随机种子是为了确保每次生成固定的随机数,这就使得每次实验结果显示一致了,有利于实验的比较和改进。 cuda gpu torch 数据 0 人点赞 上一篇:分享雷军22年前编写的代码