有什么方法可以限制使用openblas编译的pycaffe的CPU使用率?

时间:2018-11-14 11:23:33

标签: machine-learning caffe pycaffe

pycaffe是使用openblas编译的。

我的pycaffe模型在训练时占用20多个CPU内核,有什么方法可以限制它?

1 个答案:

答案 0 :(得分:0)

尝试设置环境变量

OPENBLAS_NUM_THREADS=10

之前,运行pycaffe将openblas限制为仅10个线程。