Interesting. So I guess the pip-version linked to a wrong mkl version. Hence causing the issue !
In general, I would advise to use the conda install of pytorch if you’re in a conda environment. That will make sure you don’t have such issues !
Happy this is fixed.