Question

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

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

在函数中，它加载用于对象检测的tf模型并返回结果。烧瓶服务器需要同时处理大量请求。因此，我需要使用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

Answer 2

安装tensorflow gpu将使脚本自动检测gpu。如果未检测到GPU，请检查驱动程序版本（Cuda和Cudnn）。如果没有版本不匹配或错误发生，则脚本可以识别存在的gpu，并将利用gpu运行。

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

2 个答案: