如何在Google合作实验室中导入face_recognition？

问题描述

我正在Google Colab中工作，并且已经导入了dlib

!pip install dlib
import dlib

并安装了face_recognition模块。

!pip install face_recognition

但是，我在此行出现错误：

import face_recognition
import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle
from matplotlib.patches import Circle
import numpy as np
import cv2

%matplotlib inline

错误如下：

---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-159-91aca6eea0d3> in <module>()
----> 1 import face_recognition
      2 import matplotlib.pyplot as plt
      3 from matplotlib.patches import Rectangle
      4 from matplotlib.patches import Circle
      5 import numpy as np

1 frames
/usr/local/lib/python3.6/dist-packages/face_recognition/api.py in <module>()
     24 
     25 cnn_face_detection_model = face_recognition_models.cnn_face_detector_model_location()
---> 26 cnn_face_detector = dlib.cnn_face_detection_model_v1(cnn_face_detection_model)
     27 
     28 face_recognition_model = face_recognition_models.face_recognition_model_location()

RuntimeError: Error while calling cudaGetDevice(&the_device_id) in file /tmp/pip-wheel-66glv9rf/dlib/dlib/cuda/gpu_data.cpp:201. code: 100,reason: no CUDA-capable device is detected

该怎么办？

解决方法

从错误中我可以看到您（可能）尚未启用GPU加速。

转到运行时->更改运行时-> 选择GPU

然后运行代码!pip install face_recognition

这应该安装库（和依赖项）而不会出现问题。此外，您不需要单独安装dlib。让face_recognition在它旁边构建它。