【发布时间】:2018-11-14 11:23:33
【问题描述】:
pycaffe 是用 openblas 编译的。
我的 pycaffe 模型在训练时占用了 20 多个 CPU 内核,有什么方法可以限制吗?
【问题讨论】:
标签: machine-learning caffe pycaffe
pycaffe 是用 openblas 编译的。
我的 pycaffe 模型在训练时占用了 20 多个 CPU 内核,有什么方法可以限制吗?
【问题讨论】:
标签: machine-learning caffe pycaffe
尝试设置环境变量
OPENBLAS_NUM_THREADS=10
之前运行 pycaffe 将 openblas 限制为仅 10 个线程。
【讨论】: