带着Flask的皮库达给了皮库达_驱动程序逻辑错误：cuModuleLoadDataEx

Question

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

以下是相关代码：

在库达夫拉斯克.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()

在服务器.py（仅调用函数的部分）：

^{pr2}$

在运行python cudaFlask.py时，我得到了预期的输出done，但是在启动服务器并在^{上执行POST请求时，我在烧瓶控制台上得到以下错误：

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

在mod = SourceModule...线上

我哪里出错了？ 外面有一个similar question，但还没有人回答。在

Answer 1

解决了flask中延迟加载和手动生成{}的问题（即在PyCUDA中没有{}。在

请参考this中的延迟加载flask。在

我的views.py文件：

import numpy as np
import pycuda.driver as cuda
from pycuda.compiler import SourceModule

def index():
    cuda.init()
    device = cuda.Device(0) # enter your gpu id here
    ctx = device.make_context()

    mod = SourceModule("""
        int x = 4;
    """)

    ctx.pop() # very important

    print ("done")
    return "success"