启用dlib的CUDA加速功能(OS:Window10)

0x00. Introduction

做了很久ML才发现自己的dlib还在用CPU跑

国内网上教程都太老了,现版本的dlib已经无法适配老版本的CUDA和cuDNN了,VS2015的C++11编译器好像也无法编译现版本的了,自己摸索了两天(万恶的CMake)实现了。。。。

0x01. Download

1.Visual Studio 2017 

download link
note:选择框架如下
在这里插入图片描述

2.CUDA Toolkit 11.5

download link

3.cudnn-11.5-windows-x64-v8.3.0.98

download link

4.CMake 3.23.0-rc2

download link

5.dlib (默认latest version)

可以执行: (git bash

git clone https://github.com/davisking/dlib.git

或者直接访问download link下载

0x02. Configure CUDA

1 执行CUDA Toolkit 11.5安装程序,默认安装即可

note:如果提示没有支持的VS版本则代表你的VS2017未安装,或未添加至环境变量。

2 将cuDNN压缩包的 lib include bin 目录下的文件放到CUDA对应目录下
3.系统环境变量添加如下
CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.5
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64
CUDA_BIN_PATH = %CUDA_PATH%\bin
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64

4.Path下添加
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5\bin
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.5\common\lib\x64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.5\bin\win64

检验方法
输入以下命令

cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5\extras\demo_suite
bandwidthTest.exe
deviceQuery.exe

执行结果应为 Result = PASS

0x03. Compile dlib

1. 打开CMake   Browse至如下path

在这里插入图片描述

2. configure

在这里插入图片描述
note: 都2022年了,一定要选择x64 platform

3. 注意阅读result 结果应为

在这里插入图片描述
此时你的DLIB_USE_CUDA选项应该是可选的
在这里插入图片描述
如果此处为OFF,代表前面某一步错了,你要仔细阅读CMake configure的结果,写得很详细
我遇到过得错误有:

(1)CUDA simple program 无法编译
(2)cuDNN 检测异常
(3)无admin权限 

这些问题基本上报错都很详细,网上很容易找到解决方案,但一定要针对你的error

4. 将如下两项启用

在这里插入图片描述
在这里插入图片描述

5. 再次configure
6. Generate (默认打开VS2017)
7. 选择Relese
8. 生成 - 生成解决方案

note: Generate结束后提示检测到文件修改,选择忽略

0x04. Setup dlib

1. cd至dlib目录下
2. 直接输入:
python setup.py install

note: 注意报错,你应该能看到检验CUDA和cuDNN后会显示DLIB WILL USE CUDA

0x05. Check

1. 控制台输入
python
import dlib
dlib.DLIB_USE_CUDA

输出应为True

THE END.

  • 7
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
### 回答1: dlib::logger::global_data::global_data() 372是指dlib库中的logger模块中的全局数据构造函数。在创建logger对象时,会自动调用该构造函数进行初始化。 这个构造函数的主要作用是设置全局数据相关的参数和变量。在第372行,可能是进行某种数据初始化或者赋值操作。 以dlib库为例,该库是一个功能强大的C++工具包,用于开发机器学习和计算机视觉应用。其中的logger模块主要用于日志记录和调试。在构造global_data对象时,可能会初始化一些全局变量,如日志文件路径、日志等级等。 该构造函数的实际代码可能会比较复杂,具体的操作和赋值内容需要查看dlib库的源码才能确定。 ### 回答2: dlib::logger::global_data::global_data() 是dlib库中的一个构造函数。这个构造函数的作用是初始化全局日志数据。 在dlib库中,logger用于记录和输出程序的日志信息。为每个线程维护一个logger范围,可以在开发过程中调试和跟踪代码的执行流程。 该构造函数的编号为372,表示它是dlib::logger::global_data类中的第372个构造函数。 通过调用这个构造函数,可以创建一个全局的logger数据对象。这个全局数据对象将被用作所有线程中日志记录的基础。在全局数据对象的构造函数中,可以进行一些初始化工作,例如获取日志文件路径、设置日志记录级别等。 在dlib库中,全局数据对象是唯一的,它会在程序启动时创建,并在程序结束时销毁。在多线程环境下,全局数据对象提供了线程间共享和同步日志记录的功能。 具体而言,dlib::logger::global_data::global_data()函数的实现会在全局数据对象的构造过程中调用,完成一些全局性的初始化工作,并确保线程间的日志记录操作是安全和同步的。 总而言之,dlib::logger::global_data::global_data() 372这个构造函数的作用是初始化全局日志数据,为dlib库中的日志记录提供线程间共享和同步的功能。 ### 回答3: dlib::logger::global_data::global_data() 是dlib图像处理库中的一个函数。 这个函数的作用是初始化dlib库中的全局数据,为日志系统提供必要的数据结构和配置。在dlib库中,日志系统允许用户在程序中记录信息,便于排查错误和调试。global_data() 函数在程序开始运行时被调用,用于为日志系统做准备工作。 在函数内部,初始化了一些全局变量和数据结构,用于存储日志信息。例如,该函数可能会创建一个全局的日志文件管理器,用于管理写入日志文件的操作。此外,还可能会初始化一些全局配置参数,如日志级别、日志格式等。 函数的返回值为无,即不返回任何数值。 函数命名中的 "372" 可能是指该函数在dlib库的源代码中的行数。通过行号可以方便地在代码中定位该函数的具体位置。 总结来说,dlib::logger::global_data::global_data() 函数是一个用于初始化dlib库中日志系统的函数,它在程序开始时被调用,为日志的记录和管理提供必要的基础数据结构和配置。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Shihao Weng

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值