Hey @ansh7xpex I ran into a similar problem. You are probably installing the cpu version of pytorch. You need to make sure that you are installing the pytorch+cuda version. It seems that when you check for the pytorch version, you should expect somthing like ‘1.8.0+cu111’ indicating pytorch + cuda.

I could not get conda command to work because of some connection error when downloading some files. So I ended up downloading the pytorch+cuda whl version on my computer and then using the pip command. You can download the whl from here https://download.pytorch.org/whl/cu111/torch-1.9.0%2Bcu111-cp39-cp39-win_amd64.whl 1
Here is the pip command: pip install torch-1.8.0+cu111-cp39-cp39-win_amd64.whl torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
You also need to create a new conda environment to avoid some installation errors.
I hope that helps.