成功安装 dlib 仍然 dlib.DLIB_USE_CUDA 返回 false

Question

我正在尝试将 GPU 用于@ageitgey 人脸识别 api 我在安装 dlib 的过程中成功安装并编译了 dlib 我得到了 找到 CUDA：C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1（找到合适的版本“10.1”，最低要求是“7.5”） -- 正在寻找 cuDNN 安装... -- 找到cuDNN：C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1/lib/x64/cudnn.lib -- 构建一个 CUDA 测试项目，看看你的编译器是否与 CUDA 兼容... -- 检查您是否安装了正确版本的 cuDNN。 -- 为 dlib 启用 CUDA 支持。 DLIB 将使用 CUDA -- C++11 激活。 -- 配置完成 -- 生成完成

但是现在当我跑步时 导入 dlib 打印（dlib.DLIB_USE_CUDA） 在 SPYDER 我得到 FALSE 请帮忙

预期行为

我希望 (dlib.DLIB_USE_CUDA) 返回 TRUE 我希望 CNN 人脸检测器模型能够在 GPU 上运行

当前行为

dlib.DLIB_USE_CUDA 返回 false 基于 CNN 的人脸检测器占用整个 CPU 和 RAM 我还使用 nvidia-smi 检查了 gpu 的使用情况，但没有显示任何使用情况 cuda.get_num_devices() 返回 1

复制步骤

安装 dlib https://github.com/davisking/dlib 安装 cuda 工具包和其他依赖项 运行这个 https://github.com/davisking/dlib

dlib 版本 dlib==19.18.99 dlib：https://github.com/davisking/dlib 我正在使用 Windows 10 64 位 我正在使用 SPYDER

Answer 1

以下是解决此问题的步骤

1. 卸载 face_recognition 和 dlib

pip 卸载 dlib face_recognition

1. 重新安装dlib（用cuda编译）

python setup.py install

1. 再次检查

导入 dlib

dlib.DLIB_USE_CUDA

是的