Question

我已经安装了Visual Studio 2019，Cuda 10.1和TensorFlow 2.1.0，但我仍然无法使用GPU运行人脸识别，有人可以给我完整的指南来介绍使用GPU而不是CPU的步骤。

注意：我使用的是Windows 10，我的GPU是gtx1050，我使用的是蟒蛇蜘蛛。

Answer 1

我使用 anaconda 环境通过以下步骤解决了这个问题：

cuda 10.2 已安装

蟒蛇 (3.7.7)

conda 安装 pip

conda 安装 tensorflow (最新 2.1.0)

conda 安装 tensorflow-gpu

pip install imutils

pip install opencv-python

pip install opencv-contrib-python

pip 安装 dlib

pip install face_recognition

Answer 2

首先，您应该安装tensorflow-gpu软件包而不是tensorflow。

如果正确安装了tf，则可以在deepface中的gpu中运行人脸识别。您可以使用分配内存功能对其进行测试。

#!pip install deepface
from deepface import DeepFace
DeepFace.allocateMemory()

如果一切正常，则返回“ DeepFace将在GPU上运行”消息。

除Dlib之外的所有人脸识别模型都将在tensorflow-gpu上运行。您可以通过验证功能运行人脸识别。

from deepface import DeepFace
models = ["VGG-Face", "Facenet", "OpenFace", "DeepFace", "DeepID", "Dlib"]
obj = DeepFace.verify("img1.jpg", "img2.jpg", model_name = models[0])
print(obj)

如何使用GPU运行dlib人脸识别

2 个答案: