安裝dlib的時(shí)候,如果直接使用的是pip,則是不帶有GPU版本,在調(diào)用執(zhí)行的時(shí)候,使用的是CPU
pip install dlib
如果要使用GPU的話,則需要自己編譯dlib,然后再安裝的時(shí)候設(shè)置使用CUDA
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 .
cd ..
python setup.py install --set USE_AVX_INSTRUCTIONS=1 --set DLIB_USE_CUDA=1
# 原來使用的命令已經(jīng)不能用了
# python setup.py install --yes USE_AVX_INSTRUCTIONS --yes DLIB_USE_CUDA
驗(yàn)證:
import dlib
print(dlib.DLIB_USE_CUDA)
print(dlib.cuda.get_num_devices())
下面命令已經(jīng)不能成功安裝了
python setup.py install --yes USE_AVX_INSTRUCTIONS --yes DLIB_USE_CUDA
報(bào)錯(cuò)如下時(shí):
Error: The --yes options to dlib's setup.py don't do anything since all these options
are 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-commands
or: setup.py cmd --help
error: invalid command 'USE_AVX_INSTRUCTIONS'
這個(gè)時(shí)候千萬不能心急,直接根據(jù)自己想當(dāng)然安裝,不然后面又得重新卸載,重新安裝就很麻煩了,建議可以google一下,或者csdn一下(不過都沒找到解決,都是用的老命令),或者直接去github的issues里面去找問題。
解決辦法:
# 使用這條命令安裝
python setup.py install --set USE_AVX_INSTRUCTIONS=1 --set DLIB_USE_CUDA=1
聯(lián)系客服