cuDNN raising error in Transformer execution

[jsancs]

Running the transformer tutorial raises an error executing the Cross Attention module.

I think the error is related to some mismatch in the versions of the notebook, because the error appears when executing the first block of the transformer, which in this case is the Cross Attention module.

The error is related with the cuDDN launch as you can see in the screenshot. This is the error:

[jsancs]

I have seen that it is in the cuDNN and CUDA version, in the imports section. I have resolved it in the PR linked to the issue.