python - 如何在 GPU 支持下运行 python 代码

Question

我创建了一个 flask 服务，用于接受带有相机 URL 的请求，作为在相机框架中查找对象( table 、椅子等)的参数。我已经在 flask 中编写了用于接受 POST 请求的代码。

@app.route('/rest/detectObjects', methods=['GET','POST'])
def detectObjects()
...
json_result = function_call_for_detecting_objects()
...
return

在函数中，它加载了用于对象检测的tf模型并返回结果。大量的请求需要由flask服务器同时处理。因此，我需要使用 GPU 执行该功能，因为相机访问时间和对象检测的图像处理需要大量时间和 CPU 利用率。拥有 4 GB GeForce GTX 1050 Ti/PCIe/SSE2。我怎样才能让我的 python 脚本利用 GPU 来做到这一点？

Answer 1

要在 Python 中使用 GPU，您可以使用以下可用库之一:https://www.researchgate.net/post/How_do_I_run_a_python_code_in_the_GPU

CUDA 可能适合 NVIDIA GPU。有关在 Anaconda Python 发行版中使用它的指导，请参阅 https://weeraman.com/put-that-gpu-to-good-use-with-python-e5a437168c01