Принудительная установка Dlib python только с поддержкой процессора;на машине с графическим процессором с Cuda и все - PullRequest
Новая
       99

Принудительная установка Dlib python только с поддержкой процессора;на машине с графическим процессором с Cuda и все

0 голосов
/ 25 сентября 2019

Я искал его по интернету, и все, что я могу найти, это то, что у людей возникают проблемы с установкой Dlib с Cuda.У меня точно противоположная проблема.Я могу успешно установить Dlib с поддержкой Cuda и GPU в Ubuntu (Google Colab).Но я этого не хочу.Я хочу заставить Dlib использовать CPU, так как при изменении времени выполнения на TPU графического процессора не будет.Но Dlib находит CUDA во время установки и добавляет поддержку CUDA во время установки, даже если для среды выполнения установлено значение «НЕ ГПУ».

pip install Dlib

это по умолчанию установит Dlib с cuda, так как я не знаю никаких флагов для отключения Cuda.Но даже если я попытаюсь собрать Dlib из исходного кода, например: 

!git clone https://github.com/davisking/dlib.git
%cd dlib
!mkdir build
%cd build
!cmake .. -DDLIB_USE_CUDA=0 -DUSE_AVX_INSTRUCTIONS=1 
!cmake --build .
# !sudo make install
# !sudo ldconfig
%cd ..
# !python setup.py build --no DLIB_USE_CUDA
!python setup.py install --no DLIB_USE_CUDA

, я смогу успешно выполнить 

import Dlib

Но когда я использовал «face_recogntion», которое вызывает Dlib в бэкэнде 

!pip install face_recognition
import dlib
import face_recognition

выдает ошибку: 

---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-52-c42385b122ec> in <module>()
      2 get_ipython().system('pip install face_recognition')
      3 import dlib
----> 4 import face_recognition

1 frames
/usr/local/lib/python3.6/dist-packages/face_recognition/api.py in <module>()
     21 
     22 cnn_face_detection_model = face_recognition_models.cnn_face_detector_model_location()
---> 23 cnn_face_detector = dlib.cnn_face_detection_model_v1(cnn_face_detection_model)
     24 
     25 face_recognition_model = face_recognition_models.face_recognition_model_location()

RuntimeError: Error while calling cudaGetDevice(&the_device_id) in file /tmp/pip-install-9p_kkgzg/dlib/dlib/cuda/gpu_data.cpp:178. code: 38, reason: no CUDA-capable device is detected

Теперь я не могу понять, что мне не хватает.Какой флажок следует установить, чтобы заставить DLIB использовать процессор и даже не искать графический процессор, даже если он есть.

...