Pytorch设置随机数种子

在使用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)