Free(): invalid pointer Aborted (core dumped) on loss.backward()

nlp
3 
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

I installed pytorch through pip using the command created by pytorch.org

pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

And my GPU is NVIDIA Geforce GTX 1050Ti

However, I’m not sure how to check CuDNN version. This is what I see after running the installation file- “libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb”
Screenshot from 2021-03-30 15-10-01
I have never set up a GPU device before. So I’m not sure what I’m doing to be honest.