最近因项目需要,使用dlib库,但是pip安装的dlib无法使用gpu加速功能,故查找多方面材料,折腾了几天才弄好。
参考链接:https://blog.csdn.net/luckyfairy17/article/details/83855739
Windows10下编译python版dlib
- 1. 卸载原来的dlib,pip uninstall dlib
- 2. 安装vs2015 update3,链接:http://download.microsoft.com/download/8/4/3/843ec655-1b67-46c3-a7a4-10a1159cfa84/vs2015.3.ent_enu.iso
- 3. 下载与显卡匹配的cuda和cudnn
下载cuda:https://developer.nvidia.com/cuda-80-ga2-download-archivel
然后下载cudnn:https://developer.nvidia.com/rdp/cudnn-download
这里需要注册一个nVidia账号,注册登陆之后下载对应版本即可,我下载的cudnn-8.0-windows10-x64-v7.1-ga.
配置cuda+cudnn
- 4. 安装完成之后,将cudnn中所有文件复制到cuda安装目录;
添加环境变量
首先上述安装完成之后,系统中多了
还需添加
另外:系统变量PATH的末尾添加:
注:视自己的版本为主;
- 5. 配置完成之后,验证是否成功。
cmd进入到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA \v8.0\extras\ demo_suite
运行bandwidthTest.exe
运行deviceQuery.exe
结果显示pass,就是配置成功。
- 6. 下载dlib,http://dlib.net/
最新版的是dlib19.17
解压cmd进入dlib安装文件夹,因为新版的取消了--yes命令,原来博客中--yes命令无法使用,只需要
python setup.py install
一路编译结束,过程有一点长。
- 7.Python里测试dlib有没有gpu加速
显示为True,即成功配置gpu版dlib。
另:如果出现print(dlib.DLIB_UES_CUDA)->False的情况,则是vs版本不对,只需要vs2015 update3即可
转载请标明出处:Windows10环境编译配置python版dlib19.17
文章来源: https://blog.csdn.net/weixin_42483488/article/details/96837798