win10下的使用dlib的GPU加速

时间:2024-03-15 16:44:35

发现上文已经配置好了dlib的环境,默认是有GPU加速功能的。

在cmake的时候有显示 Enabling CUDA support for dlib. DLIB WILL USE CUDA

之类的字眼,就是支持GUP加速。

本地已经安装了

1.CUDA 9.0

2.显卡是P2000

3.下载CUDNN,将解压出来的cuda文件放在duda9.0下。

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

4.打开VS2015,添加cuda和dlib的lib文件路径和文件

dlib.lib

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64

依自己的cuda和cudnn目录修改

win10下的使用dlib的GPU加速

cudnn.lib
cudart_static.lib
cublas.lib
cublas_device.lib
curand.lib
cusolver.lib
dlib.lib

win10下的使用dlib的GPU加速

最后在预处理添加DLIB_USE_CUDA

win10下的使用dlib的GPU加速

 

补充:python下验证GPU加速

# python script

import dlib

print(dlib.__)

print(dlib.DLIB_USE_CUDA)

print(dlib.cuda.get_num_devices())