时间戳

首页 > 解决方案 > 如何在 GPU 的支持下运行 python 代码

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

问题描述

我创建了一个烧瓶服务，用于接受带有摄像头 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？

标签： pythonflaskartificial-intelligencegpu

解决方案

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

推荐阅读