It seems that a cudnnHostCalloc call is failing. How did you install PyTorch and which cudnn version are you using?
Since you mentioned CUDA11.2, did you build PyTorch from source or are you using the binaries?
Also, which GPU are you using?