Solved by albanD in post #17 
              
              
                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.