Reproducibility in C++

C++
1

The PyTorch docs say that you need to set torch.backends.cudnn.deterministic = True for reproducible runs.

What’s the equivalent setting in the C++ front-end?

2

Hi Thomas,

This is part of the globalContext (ATen/Context.h): Would at::globalContext().setDeterministicCuDNN(true) help?

Best regards

Thomas

1 Like