首页 > 解决方案 > 如何使用 gpu 运行 dlib 人脸识别

问题描述

我已经安装了 Visual Studio 2019、Cuda 10.1 和 TensorFlow 2.1.0,但我仍然无法使用 GPU 运行人脸识别,有人可以给我一份关于使用 GPU 代替 CPU 步骤的完整指南。

注意:我使用的是 Windows 10,我的 GPU 是 gtx1050,我使用的是 anaconda spider。

标签: tensorflowgpuface-recognitiondlib

解决方案


我使用 anaconda 环境通过这些步骤解决了这个问题:

已安装 cuda 10.2

蟒蛇(3.7.7)

康达安装点子

康达安装张量流(最新的2.1.0)

conda 安装 tensorflow-gpu

点安装 imutils

点安装 opencv-python

点安装 opencv-contrib-python

点安装 dlib

pip install face_recognition


推荐阅读