Я хочу, чтобы запустить код PyCuda на flask
сервере. Файл работает правильно, используя 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...
Где я иду не так? Аналогичный вопрос есть, но ответа еще не получено.