seed = 2023
torch.manual_seed(seed) # torch的CPU随机性,为CPU设置随机种子
torch.cuda.manual_seed(seed) # torch的GPU随机性,为当前GPU设置随机种子
torch.cuda.manual_seed_all(seed) # torch的GPU随机性,为所有GPU设置随机种子
标签:设置,torch,manual,seed,种子,随机数,深度,GPU
From: https://www.cnblogs.com/fly-smart/p/17780167.html