dlib обучает нейронную сеть на процессоре

Может быть это глупый вопрос, но я некоторое время искал в Интернете, и я не нашел ответа.

Есть ли возможность обучать нейронную сеть (dlib) на процессоре, когда библиотека построена с поддержкой CUDA? Если да, то как это сделать?


gpu cpu dlib neural-network
person Michal    schedule 05.07.2018    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Нет. Когда он будет создан для использования CUDA, он будет использовать CUDA. Если вы хотите использовать ЦП, вам нужно перекомпилировать с выключенным CUDA. Тем не менее, вы определенно не хотите этого делать. Это намного медленнее на процессоре.

person Davis King    schedule 05.07.2018
comment
Я задаю этот вопрос, потому что иногда обучение нейронной сети дает сбой, и я не знаю, почему. Ошибка, например, в вызове cudaMemcpy с параметром cudaMemcpyDeviceToHost, где причина: обнаружен незаконный доступ к памяти, но если я снова запускаю обучение (загружаю последнее сохраненное состояние нейронной сети из файла синхронизации), оно продолжается без каких-либо ошибок, или ошибка появляется снова в другой момент обучения. В чем проблема? Может с железом что-то не так? - person Michal; 07.07.2018