PyTorch Forums
Manually set cudnn convolution algorithm
vision
ptrblck
September 24, 2021, 6:54pm
6
A general cuDNN extension example can be found
here
and
this post
might also be useful.
show post in topic