相关疑难解决方法(0)

我想在flask服务器上运行pyCUDA代码.文件直接正确运行python3但在使用时调用相应的函数失败flask.

这是相关代码:

cudaFlask.py:

import pycuda.autoinit
import pycuda.driver as drv
import numpy

from pycuda.compiler import SourceModule

def cudaTest():
    mod = SourceModule("""
        int x = 4;
    """)

    print ("done")
    return

if __name__ == "__main__":
    cudaTest()

server.py(只调用该函数的部分):

@app.route('/bundle', methods=['POST'])
def bundle_edges():
    cudaTest()
    return "success"

在运行时,python cudaFlask.py我得到了done预期的输出,但在启动服务器并执行POST请求时,website/bundle我在烧瓶控制台上收到以下错误:

pycuda._driver.LogicError: cuModuleLoadDataEx failed: invalid device context - 

在线上 mod = SourceModule...

我哪里错了？有一个类似的问题,但尚未得到回答.