import numpy as np import os import random import torch def seed_all(seed): np.random.seed(seed) os.environ['PYTHONHASHSEED'] = str(seed) random.seed(seed) torch.manual_seed(seed) torch.cuda.manual_seed(seed) torch.backends.cudnn.deterministic = True