python setup.py install
前提是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版本安装成功
原文:https://www.cnblogs.com/jaysonteng/p/14846433.html