How to use AUTO_TUNE with pytorch?

How to use NVIDIA AUTO TUNE with pytorch.

Hi,

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