scipy.linalg.sqrtm get stuck in SLURM clsuters

My issue is about the scipy.linalg.sqrtm getting stuck in SLURM cluster. I do not know why, when I run scipy.linalg.sqrtm in the cluster, the program will get stuck in running. But It runs well on my personal computer

Reproducing code example:

    covmean, _ = linalg.sqrtm(sigma1.dot(sigma2), disp=False,blocksize=32)
    sigma1 and sigma2 are two 2048*2048 matrix

Error message:

    the procedure will get stuck at "  covmean, _ = linalg.sqrtm(sigma1.dot(sigma2), disp=False,blocksize=32)"

Scipy/Numpy/Python version information:

 scipy     1.7.0+computecanada
numpy       1.20.1
python 3.7
 So please help me! thank you!

[LonglongaaaGo]

Hi, all！@ilayn @ev-br @czgdp1807 @brandon-beacher Thanks for your help! Now, I solved this problem. By setting ‘export MKL_NUM_THREADS=1’, my codes can run well. It seems there is some weird interaction between torch and scipy.

[rimless]

thank you, I met the same problem