时间戳

首页 > 解决方案 > 成功安装 dlib 仍然 dlib.DLIB_USE_CUDA 返回 false

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

问题描述

我正在尝试将 GPU 用于 @ageitgey 人脸识别 api 我在安装 dlib 期间成功安装并编译了 dlib 我得到了 Found 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 已激活。-- 配置完成 -- 生成完成

但是现在当我在 SPYDER 中运行 import dlib print(dlib.DLIB_USE_CUDA) 时，我得到 FALSE 请帮忙

预期行为

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

当前行为

dlib.DLIB_USE_CUDA 返回错误 基于 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

标签： face-recognitiondlib

解决方案

以下是解决此问题的步骤

  1. 卸载 face_recognition 和 dlib

pip 卸载 dlib face_recognition

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

python setup.py 安装

  1. 再检查一遍

导入 dlib

dlib.DLIB_USE_CUDA

真的

推荐阅读