dlib安装-CPU-GPU


conda安装

conda install dlib  # cpu版本

编译安装

CPU版本

  • 下载链接:http://dlib.net/files/
    注意:python和dlib有版本兼容问题,需要下载兼容的版本,不然安装会报错
  • 安装cmake: pip install cmake
  • 下载后,解压,进入目录后,执行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版本安装成功

参考链接


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM