torch np cuda seed random everything

 

refer to below code:

..

import numpy as np

import os

import random

import torch

SEED = 42

def seed_everything(seed):

random.seed(seed)

os.environ['PYTHONHASHSEED'] = str(seed)

np.random.seed(seed)

torch.manual_seed(seed)

torch.cuda.manual_seed(seed)

torch.cuda.manual_seed_all(seed)

torch.backends.cudnn.deterministic = True

seed_everything(SEED)

..

Thank you.

www.marearts.com