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