ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

dlib gpu版本的安装

2021-07-09 15:31:07  阅读:526  来源: 互联网

标签:USE python CUDA 版本 gpu 安装 dlib


CPU版本安装

  • 下载链接:http://dlib.net/files/
    注意:python和dlib有版本兼容问题,需要下载兼容的版本,不然安装会报错
  • 下载后,解压,进入目录后,执行python setup.py install

GPU版本的安装

前提是CUDA和CUDNN需要正确安装,不然即使后续流程正确,也无法使用到GPU

git clone https://github.com/davisking/dlib.git 
cd dlib 
mkdir build 
cd build 
cmake .. -DDLIB_USE_CUDA=1 -DUSE_AVX_INSTRUCTIONS=1
cmake --build . # 注意有个'.'点号。cuda10无需此命令,cuda9 会报log1p不允许错误,将dlib/dlib/cuda/cuda_dlib.cu 第1691行 log1p(x)改成log(1+x)
cd ..
python setup.py install --set USE_AVX_INSTRUCTIONS=yes --set DLIB_USE_CUDA=yes
# 验证dlib
python -c 'import dlib; print(dlib.DLIB_USE_CUDA)'  # 如果输出True,表示GPU版本安装成功

参考链接

标签:USE,python,CUDA,版本,gpu,安装,dlib
来源: https://www.cnblogs.com/jaysonteng/p/14846433.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有