| import torch | |
| import numpy as np | |
| import random | |
| seed = 0 | |
| torch.manual_seed(seed) | |
| torch.cuda.manual_seed(seed) | |
| torch.cuda.manual_seed_all(seed) # if you are using multi-GPU. | |
| np.random.seed(seed) # Numpy module. | |
| random.seed(seed) # Python random module. | |
| torch.manual_seed(seed) | |
| torch.backends.cudnn.benchmark = False | |
| torch.backends.cudnn.deterministic = True |