Успешно установленный dlib все еще dlib.DLIB_USE_CUDA возвращает false - PullRequest
1 голос
/ 07 октября 2019

Я пытаюсь использовать GPU для API распознавания лиц @ageitgey. Я успешно установил и скомпилировал dlib во время установки dlib. Найдено CUDA: C: / Program Files / NVIDIA GPU Computing Toolkit / CUDA / v10.1 (найдена подходящая версия«10,1», минимальное обязательное значение «7,5») - Требуется установка cuDNN ... - Найден cuDNN: C: / Program Files / NVIDIA GPU Computing Toolkit / CUDA / v10.1 / lib / x64 / cudnn.lib -- Создание тестового проекта CUDA для проверки совместимости вашего компилятора с CUDA ... - Проверка, установлена ​​ли у вас правильная версия cuDNN. - Включение поддержки CUDA для dlib. DLIB будет использовать CUDA - C ++ 11 активирован. - Конфигурирование завершено - Генерация завершена

НО теперь, когда я запускаю импорт dlib print (dlib.DLIB_USE_CUDA) в SPYDER Я получаю FALSE, пожалуйста, помогите

Ожидаемое поведение

Я ожидал, что (dlib.DLIB_USE_CUDA) вернет TRUE. Я ожидал, что модель детектора лиц CNN будет работать на графическом процессоре

Текущее поведение

dlib.DLIB_USE_CUDA возвращает ложное Детектор лица на основе CNN забирает весь ЦП и ОЗУЯ также проверил использование графического процессора с помощью nvidia-smi, но он не показывал никакого использования. Cuda.get_num_devices () возвращает 1

Шаги для воспроизведения

install dlib https://github.com/davisking/dlib install cuda toolkitи другие зависимости Запустите эту https://github.com/davisking/dlib

версию dlib dlib == 19.18.99 dlib: https://github.com/davisking/dlib Я работаю в Windows 10 64 бит Я использую SPYDER

...