dlib cuda不使用GPU

Question

我想通过使用Nvidia GPU而不是CPU来改进我的CNN面部识别代码。因此，我在these instructions之后找到并安装了特定的dlib_cuda

安装进展顺利，因此我检查了dlib是否在Python环境中使用cuda：

Python 3.6.9 (default, Jul  17 2020, 12:50:27)
[GCC 8.4.0] on linux
>>> import dlib
>>> dlib.DLIB_USE_CUDA
True
>>>print(dlib.cuda.get_device())
1

因为它看起来对我很好，我尝试再次使用我的代码，但没有任何改进，在检查GPU后，它仍然没有使用。因此，我尝试了以下命令：

>>> print(dlib.cuda.get_device())

它返回：

0

我不知道这些信息是什么意思。经过大量的研究，我仍然无法理解dlib没有使用我的GPU。以前有人面临过同样的问题吗

我的工作区在Jetson AGX Xavier（Jetpack 4.4）上，运行Ubuntu和Cuda版本10.2.89

PS：我还使用tensorFlow和Keras库，它们都是为配合CUDA环境而安装的

Answer 1

如果您熟悉Tensorflow和Keras，我建议您使用deepface进行人脸识别。它包装了最先进的人脸识别模型，并在后台构建了Keras模型。这就是为什么，如果您安装了tensorflow GPU包，它的默认用法将在GPU上执行

#!pip install deepface
from deepface import DeepFace
models = ["VGG-Face", "Facenet", "OpenFace", "DeepFace", "DeepID"]
resp = DeepFace.verify("img1.jpg", "img2.jpg", model_name = models[0])
print(resp["verified"])