发现上文已经配置好了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目录修改
cudnn.lib
cudart_static.lib
cublas.lib
cublas_device.lib
curand.lib
cusolver.lib
dlib.lib
最后在预处理添加DLIB_USE_CUDA
补充:python下验证GPU加速
# python script
import dlib
print(dlib.__)
print(dlib.DLIB_USE_CUDA)
print(dlib.cuda.get_num_devices())