当前位置:   article > 正文

【全网第一详细】Windows下编译使用GPU版本的DLib_windows 安装dlib gpu

windows 安装dlib gpu

一、准备基础环境

因为要用上GPU所以提前准备好:

  1. CUDA

将cuda/bin和cuda/lib/x64以及cuda/include添加到环境变量Path

  1. CUDNN

解压CUDNN然后把它对应文件夹中的文件添加到CUDA对应的文件夹目录下

可以参考:深度学习GPU环境CUDA详细安装过程(简单快速有效)

二、下载Dlib官方包

https://github.com/davisking/dlib/releases

下载最新版本即可

三、安装cmake

Windows下CMake安装教程

四、编译Dlib

管理员模式cmd进入到Dlib文件夹下,输入:

python setup.py install
  • 1

等待编译完成!

五、嵌入到Python中使用(重要)

之前我一直不知道怎么编译好使用,直到我看到这篇文章 ,里面作者详细说明了在Anaconda3中编译使用Dlib,经过我的实践确实可以用,他有一个关键的操作就是将编译的Dlib放到Anaconda3的目录下,所以我觉得在Python中使用应该也是如此,于是我:

把编译好的Dlib整个文件夹放入到了 E:\python365\Lib\site-packages 下(放到你的python安装路径下)
  • 1

六、测试使用

import dlib
print(dlib.DLIB_USE_CUDA)
print(dlib.cuda.get_num_devices())
  • 1
  • 2
  • 3
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/正经夜光杯/article/detail/819672
推荐阅读
相关标签
  

闽ICP备14008679号