如何做并行Python Gekko？

Question

有时，一个求解器比另一个求解器能更好地解决我的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))

Answer 1

您可以使用多线程应用程序同时在任意数量的服务器上进行求解。下面是使用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)