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

Question

我正在尝试将图形处理器用于@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激活。--配置完成--生成完成

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

预期行为

我期望(dlib.DLIB_USE_CUDA)返回TRUE我期望CNN人脸检测器模型在图形处理器上运行。

当前行为

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

重现步骤

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

dlib版本SPYDER：https://github.com/davisking/dlib我正在Windows1064位上工作，我正在使用SPYDER

Answer 1

以下是解决此问题的步骤

1. 卸载face_recognition和dlib

pip卸载dlib face_recognition

python setup.py安装

1. check again

导入dlib

dlib.DLIB_USE_CUDA

真的