将 MKL 优化 scipy 限制为单线程
Restrict MKL optimized scipy to single thread
我刚刚安装了 scipy 的英特尔 MKL 优化版本,当 运行 在我的基准测试中,我得到了显着的加速。然后我仔细观察,发现它在 20 个内核上 运行ning ... 我如何将它限制为单线程模式?有没有一种方法可以默认将它安装到单线程模式,同时在指定数量的内核上将选项打开为 运行?
经进一步调查,您似乎可以设置环境变量 MKL_NUM_THREADS
来实现此目的。
我刚刚安装了 scipy 的英特尔 MKL 优化版本,当 运行 在我的基准测试中,我得到了显着的加速。然后我仔细观察,发现它在 20 个内核上 运行ning ... 我如何将它限制为单线程模式?有没有一种方法可以默认将它安装到单线程模式,同时在指定数量的内核上将选项打开为 运行?
经进一步调查,您似乎可以设置环境变量 MKL_NUM_THREADS
来实现此目的。