文章更新日志:
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
安装完成,配置系统环境变量(选择自己的路径)
D:\Develop\CMake\bin
3、下载安装CUDA
选择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解压到指定目录下,我这里目录是 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 命令
进入 Dlib目录,执行目录安装
如: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,但是失败了
(如果你成功了,请评论区给出解决办法,十分感谢!)
https://github.com/davisking/dlib
总结:
本次配置消耗了我两天的时间(来之不易的周末就这样没了),配置失败的主要原因就是Visual Studio版本太高了,导致版本间不匹配;其次就是没有好好阅读Dlib官方的文档(英语阅读能力不行,导致错过太多关键步骤)。
建议:
1、多看官方提供的配置流程,比很多博客管用
2、如果需要看博客参考,建议查看比较新的博客,因为一般配置的环境都用比较新的,找新的博客能避免很多不必要的麻烦
3、多锻炼英语阅读能力(非常重要)