CUDNN Setting

17 May 2021

Reproducibility

True

import numpy as np
np.random.seed(0)
torch.manual_seed(0)
torch.cuda.manual_seed_all(0)

torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False

False

torch.backends.cudnn.benchmark = True