(python dlib调用GPU(win10+visual studio2019回顾)(cmake编译))
用GPU算图像啥的这些张量速度确实比CPU快了n倍,之前直接pip install的dlib,但是跑的速度简直堪比乌龟,查了下才知道是dlib没有调用GPU。那GPU不是白买的,实在忍不了,花了整整一天才配好,中间踩了很多坑,记录一下。这是我的第一篇博客,我就是个渣渣,但是如果这篇博客能帮到别人,我也会很开心。
运行环境
win10 + visual studio2019 + cuda +cudnn
cuda和cudnn但凡调用显卡那基本上是必须的,这几个软件可以自己下,下载安装的教程网上有很多,这里不再赘述了。
我编译dlib主要参考的是这篇博客:https://www.cnblogs.com/supersayajin/p/8446685.html
配置过程在第四步之前都没啥问题,但是在camke的时候,问题来了。先是configure的时候找不到CUDA,实在没办法把cuda重装了一次,选简易安装,然后照那篇博客说的把cudnn的几个文件夹下面的文件对应复制,其实cuda10.0不用这么做,几个文件都有。然后cmake好不容易找到cuda了,但是又报错
CUDA was found but your compiler failed to compile a simple CUDA program so dlib isn’t going to use CUDA.
– *** The output of the failed CUDA test compile is shown below:
– *** Change Dir: F:/Program Files(x86)/Anaconda/dlib-19.14.0/build/temp.win-amd64-3.7/Release/dlib_build/cuda_test_build
查了下,是vs版本在代码里面被限制住了。按照这个**[网页]**(https://blog.zhazhafang.cn/?p=177)上面的办法,管理员运行程序改了一下,总算成了。**对了,Windows下一定要在configure之后的界面里面把vs版本和平台版本(x32orx64)选好。**that’s all