如何在cvxpy中设置线程的限制数量

时间:2019-03-14 04:20:17

标签: cvxpy

我在共享计算机上使用CVXPY,我必须设置有限数量的线程。

prob = cvx.Problem(objective, constraints)
prob.solve(solver=cvx.CVXOPT)

是否有任何选择来限制CVXPY解算器的线程数?

谢谢!

1 个答案:

答案 0 :(得分:0)

更新该问题的解决方案。默认情况下,CVXPY和NUMPY都可以根据可用内核创建线程。您可以选择在导入NUMPY和CVXPY之前限制OMP_NUM_THREADS。

import os
os.environ["OMP_NUM_THREADS"] = "1"