Как решить cudaSetDevice Error и правильно запустить этот углубленный код?

Question

Я использую следующий код GPU для распознавания выражений лица: http://code.google.com/p/deep-learning-faces/

Моя система такая же, как у авторов, но версия CUDA немного выше (5.5), чем у авторов (5.0).

Когда я запускаю "script_face_exp.m", он говорит, что "Ошибка при использовании mexcuConvNNoo Assertion Failed: cudaSetDevice Error!". Это вызвано: cudaErrorInvalidDevice Это указывает на то, что порядковый номер устройства, предоставленный пользователем, не соответствует действительному устройству CUDA.

Но я могу использовать GPU для вычислений, и коды, основанные на theano, хорошо работали на GPU.

Как я могу решить использовать графический процессор для правильного выполнения кода?

Answer 1

script_face_exp функция вызова gg который устанавливает эти глобальные переменные:

function [] = gg( gpunumber )
global GPUID;
global FERMI;
GPUID = 3;
FERMI = 1;

Глобальная переменная GPUID позже используется в SelectGPUFromServer установить используемый графический процессор.

Вероятно, у вас нет графического процессора с идентификатором 3 (что означает наличие как минимум 4 установленных графических процессоров), поэтому вам следует изменить gg чтобы установить эту переменную, например, 0 (если у вас есть один графический процессор).