Dlib Gpu 版本成功编译经历

最新推荐文章于 2025-01-26 19:05:33 发布

原创最新推荐文章于 2025-01-26 19:05:33 发布 · 469 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#python

本文详细介绍了如何在VisualStudio2015环境下成功编译dlib19.18并启用CUDA9支持。关键步骤包括使用cmake指定VisualStudio2015为编译器，设置DDLIB_USE_CUDA和USE_AVX_INSTRUCTIONS选项，并修改setup.py文件以确保正确构建。最终通过pythonsetup.pyinstall命令完成安装。

部署运行你感兴趣的模型镜像

成功编译配置: dlib 19.18、 cuda9、vs2015
附上成功截图:
在这里插入图片描述
vs2015 是关键，其他的版本编译都会出错，显示不使用 cuda!!!##

安装cuda 和 vs2015 步骤省略，自己参考其他的博客
进入到dlib 目录

mkdir build
cd build
cmake .. -G "Visual Studio 14 2015 Win64" -T host=x64  -DDLIB_USE_CUDA=1 -DUSE_AVX_INSTRUCTIONS=1 # 这里是选择vs2015 作为编译器
cmake --build .

修改 setup.py 文件将151 行修改和下面一样

151：cmake_args += ['-T', 'host=x64']

输入编译指令

python setup.py install --set USE_AVX_INSTRUCTIONS=1 --set DLIB_USE_CUDA=1 -G "Visual Studio 14 2015 Win64"

编译成功

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理