Thanks for the update.
As a workaround for now, you could disable cudnn via torch.backends.cudnn.enabled = False and try to run the code.
Could you also post the model definition, all arguments, as well as the input shapes to reproduce this error, please?