在使用PyTorch时,如果希望通过设置随机数种子,在gpu或cpu上固定每一次的初始化种子,从而使代码结果可复现,可在程序执行的开始处添加以下代码:
def setup_seed(seed):
random.seed(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed)
torch.backends.cudnn.benchmark = False
torch.backends.cudnn.deterministic = True
setup_seed(args.seed)