Tensorflow 如何使用gpu运行dlib人脸识别

我已经安装了visual studio 2019、Cuda 10.1和TensorFlow 2.1.0，但我仍然无法使用GPU运行人脸识别，有人能给我一个完整的指南，告诉我如何使用GPU而不是CPU

---

注意：我使用的是windows 10，我的GPU是gtx1050，我使用的是anaconda spider。

首先，你应该安装tensorflow GPU包而不是tensorflow

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

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

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)

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

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

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

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)

---

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

cuda 10.2已安装

python（3.7.7）

康达安装pip

康达安装tensorflow（最新2.1.0）

康达安装tensorflow gpu

pip安装imutils

pip安装opencvpython

pip安装opencv contrib python

pip安装dlib

pip安装人脸识别