How to use AUTO_TUNE with pytorch?

How to use NVIDIA AUTO TUNE with pytorch.


I am not familiar with this? Could you explain what it is and what it does please?

I’m not familiar with TensorFlow, but it seems their “auto tune” functionality corresponds to torch.backends.cudnn.benchmark = True mode.

1 Like