python - 如何做并行Python Gekko？

有时我的Python Gekko应用程序在一个解算器上比另一个解算器上解决得更好。很难预测哪个解算器的性能最好。因为Python Gekko支持在不同服务器上使用m.GEKKO(server='http://{address}')进行本地或远程求解，所以是否可以创建一个并行的Gekko应用程序，该应用程序将在任意数量的计算机（包括本地计算机）上同时尝试所有解算器，然后在第一个进程成功返回时终止其他进程？我一直在研究Python的多线程和并行包。有没有什么能很好地与Gekko进行并行求解的呢？下面是一个连续的原型：

from gekko import GEKKO
m = GEKKO()
x = m.Var(); y = m.Var()
m.Equation(x**2+y**2==1)
m.Maximize(x+y)
# try all solvers
for i in range(1,4):
    m.options.SOLVER = i
    m.solve()
    if m.options.APPSTATUS==1:
        print('Success: ' + str(i))

最佳答案

您可以在多线程应用程序同时解决任意数量的服务器上的问题。这是一个使用queue的串行应用程序的并行版本，尽管它在第一个线程完成时不会停止其他线程。即使其他线程没有停止，您也可以包含一个MAX_TIME value，以便强制线程在指定的秒数内完成。这种方法允许您继续主程序，同时让其他线程在它们完成或达到最大时间限制时死亡。

from gekko import GEKKO
import queue, threading

def solve(rq, solver, server):
    #print('Solver ' + str(solver) + ' Server: ' + str(server) + '\n')
    if server=='localhost':
        m = GEKKO(remote=False)
    else:
        m = GEKKO(server=server)
    x = m.Var(); y = m.Var()
    m.Equation(x**2+y**2==1)
    m.Maximize(x+y)
    m.options.SOLVER = solver
    m.options.MAX_TIME = 2
    m.solve(disp=False)
    m.cleanup()
    rq.put(('solver',solver,'server',server))

# Select servers and solvers (1=APOPT, 2=BPOPT, 3=IPOPT, etc)
APOPT = 1; BPOPT = 2; IPOPT = 3
jobs = {'https://byu.apmonitor.com':APOPT,\
        'localhost':APOPT,\
        'https://byu.apmonitor.com':IPOPT,\
        'https://apmonitor.com':IPOPT}

# Run jobs, get only first result
q = queue.Queue()
threads = [threading.Thread(target=solve, args=(q, solver, server)) \
           for server, solver in jobs.items()]
for th in threads:
    th.daemon = True
    th.start()
first = q.get(); print(first)

如果您还想获得第二个或更多结果，那么您可以添加另一行的q.get()个呼叫。

second = q.get(); print(second)