windows10环境下，Python 使用dlib cuda 编译和安装

codeの诱惑

已于 2024-01-22 18:35:47 修改

阅读量1.5k

点赞数 25

文章标签： python 开发语言

于 2024-01-21 23:53:58 首次发布

本文链接：https://blog.csdn.net/qq_45145827/article/details/135736991

版权

文章更新日志：
2024-01-22 创建文章
2024-01-22 修改：因为cuda版本问题，文章内容需要修正，主要核对【环境需要的版本】里建议的即可

为了方便下载，我把资源单独打包再一起了（cuda文件太大了，上传不了，去官网找吧）
下载链接

忙碌了一天，终于把基于CUDA的dlib搭建起来了，分享一下搭建过程。

环境需要的版本：

Visual Studio Community 2019 16.x
CMake 3.28.1
cuda_10.2
cudnn 8.0.2.39
Dlib 19.21

1、下载安装Visual Studio

Visual Studio Community 2019 V16.x

打开后，重点选择这三个选项安装
在这里插入图片描述
安装完成后就可以退出了。

2、下载安装CMake

CMake 3.28.1

安装完成，配置系统环境变量（选择自己的路径）

D:\Develop\CMake\bin

3、下载安装CUDA

cuda_10.2.89_441.22_win10.exe

选择exe标签后，点击Download按钮下载并安装
在这里插入图片描述

4、下载Cudnn压缩包

cudnn-10.2-windows10-x64-v8.0.2.39.zip

点击下载压缩包
在这里插入图片描述

重点：

(1)、解压后，复制 lib、include 、bin 这三个目录

(2)、进入 NVIDIA 的安装位置，比如我的路径是：
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
然后粘贴，把 CUDA的 lib、include 、bin 给替换掉

(3)、配置NVIDIA的系统环境变量，在 Path 路径

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\include

(4)、新增一条 Cmake 环境变量
变量名：CMAKE_PREFIX_PATH
路径：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1

注：如果这个没替换好，会导致Dlib编译时找不到 NVIDIA 的配置

5、下载Dlib库

Dlib官网
 Dlib-19.21下载链接

以上环境安装配置好以后，把Dlib解压到指定目录下，我这里目录是 D:\win\anaconda3\envs\pptool\etc\dlib-19.21
打开CMD，进入这个目录里面输入以下命令

mkdir build
cd build
cmake ..
cmake --build . --config Release

(6)、进入Cmake设置CUDA配置

打开Cmake应用

在这里插入图片描述
编译后，出现 Enabling CUDA support for dlib. DLIB WILL USE CUDA, compute capabilities: 50 说明Cuda打开成功了

(7)、执行 python setup.py install 命令

如：D:\win\anaconda3\envs\pptool\etc\dlib-19.24

python setup.py install

结果：

Installed d:\win\anaconda3\envs\pptool\lib\site-packages\dlib-19.24.0-py3.9-win-amd64.egg
Processing dependencies for dlib==19.24.0
Finished processing dependencies for dlib==19.24.0

(8)、验证结果

依次输入

python
import dlib
print(dlib.DLIB_USE_CUDA)

输出结果为True 说明配置成功了

问题1：

 unsupported Microsoft Visual Studio version! Only the versions between 2017 and 2019 (inclusive) are supported! The nv
cc flag '-allow-unsupported-compiler' can be used to override this version check; however, using an unsupported host co

出现这个提示，说明这个Dlib版本支持 Visual Stadio 的版本是在 2017~2019之间，如果有多个版本，建议把高出 2019版本的给卸载掉

当然我也尝试过官方建议的指定版本安装Dlib，但是失败了
（如果你成功了，请评论区给出解决办法，十分感谢！）