安裝dlib gpu版本記錄(解決遇到的Error: The --yes options to dlib‘s setup.py don‘t do anything……)
阿新 • • 發佈:2020-12-11
安裝dlib cpu
安裝dlib的時候,如果直接使用的是pip,則是不帶有GPU版本,在呼叫執行的時候,使用的是CPU
pip install dlib
安裝dlib gpu
如果要使用GPU的話,則需要自己編譯dlib,然後再安裝的時候設定使用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
# 原來使用的命令已經不能用了
# python setup.py install --yes USE_AVX_INSTRUCTIONS --yes DLIB_USE_CUDA
驗證:
import dlib
print(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 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'
這個時候千萬不能心急,直接根據自己想當然安裝,不然後面又得重新解除安裝,重新安裝就很麻煩了,建議可以google一下,或者csdn一下(不過都沒找到解決,都是用的老命令),或者直接去github的issues裡面去找問題。
解決辦法:
# 使用這條命令安裝
python setup.py install --set USE_AVX_INSTRUCTIONS=1 --set DLIB_USE_CUDA=1