配置信息
本教程默认大家都已装有cuda和cudnn~
本人cuda版本为10.0,cudnn版本为7.6.5,注意cuda版本为10.0时,cudnn对应版本必须只是高于7.5才能编译dlib的GPU版本,而且VS的版本必须为2015或2017(我选的版本),其它版本不行。
安装步骤
下载dlib并解压
登录dlib官网
下载cmake
用conda或者pip下载
conda install cmake
或者
pip install cmake
Cmake编译
- 打开cmd并在dlib文件夹创建build文件夹
mkdir build
2.在build文件夹使用cmake编译
cd build
cmake .. -G "Visual Studio 15 2017 Win64" -T host=x64
3.返回dlib文件夹运行setup.py文件安装dlib
cd ..
python setup.py install
测试代码
import dlib
dlib.DLIB_USE_CUDA
dlib.cuda.get_num_devices()
安装成功~