本文共 1195 字,大约阅读时间需要 3 分钟。
安装dlib的时候,如果直接使用的是pip,则是不带有GPU版本,在调用执行的时候,使用的是CPU
pip install dlib
如果要使用GPU的话,则需要自己编译dlib,然后再安装的时候设置使用CUDA
git clone https://github.com/davisking/dlib.gitcd dlibmkdir buildcd buildcmake .. -DDLIB_USE_CUDA=1 -DUSE_AVX_INSTRUCTIONS=1cmake --build .cd ..python setup.py install --set USE_AVX_INSTRUCTIONS=1 --set DLIB_USE_CUDA=1# 原来使用的命令已经不能用了# python setup.py install --yes USE_AVX_INSTRUCTIONS --yes DLIB_USE_CUDA
验证:
import dlibprint(dlib.DLIB_USE_CUDA)print(dlib.cuda.get_num_devices())
下面命令已经不能成功安装了
python setup.py install --yes USE_AVX_INSTRUCTIONS --yes DLIB_USE_CUDA
报错如下时:
Error: The --yes options to dlib's setup.py don't do anything since all these optionsare on by default. So --yes has been removed. Do not give it to setup.py.
博主自己去掉了yes,然后重新跑该命令也不行
usage: setup.py [global_opts] cmd1 [cmd1_opts] [cmd2 [cmd2_opts] ...]or: setup.py --help [cmd1 cmd2 ...]or: setup.py --help-commandsor: setup.py cmd --helperror: invalid command 'USE_AVX_INSTRUCTIONS'
这个时候千万不能心急,直接根据自己想当然安装,不然后面又得重新卸载,重新安装就很麻烦了,建议可以google一下,或者csdn一下(不过都没找到解决,都是用的老命令),或者直接去github的issues里面去找问题。
解决办法:# 使用这条命令安装python setup.py install --set USE_AVX_INSTRUCTIONS=1 --set DLIB_USE_CUDA=1
具体可以看
转载地址:http://zjxen.baihongyu.com/