1. 程式人生 > 其它 >安裝dlib gpu版本記錄(解決遇到的Error: The --yes options to dlib‘s setup.py don‘t do anything……)

安裝dlib gpu版本記錄(解決遇到的Error: The --yes options to dlib‘s setup.py don‘t do anything……)

技術標籤:安裝、配置問題集錦dlib gpu

安裝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

具體可以看dlib-GitHub:yes not accepted as a command ? #1839