Cntk не может обнаружить мой графический процессор cntk.all_devices ()

Cntk обнаруживает только 1 устройство (мой процессор), вызывая cntk.all_devices (). Однако на моем компьютере есть графический процессор. Запустив учебник, поддерживаемый cntk, я смог получить некоторую информацию:

-------------------------------------------------------------------
-------------------------------------------------------------------
GPU info:

                Device[0]: cores = 48; computeCapability = 2.1; type = "NVS 310"; memory = 512 MB
-------------------------------------------------------------------

##############################################################################
#                                                                            #
# Train command (train action)                                               #
#                                                                            #
##############################################################################


Model has 9 nodes. Using CPU.

Как следствие, я не могу использовать свой gpu as, вызывая set_default_device(gpu(0)). Как я мог решить эту проблему?


person freshomer    schedule 09.02.2017    source источник


Ответы (1)


Минимальная вычислительная мощность графического процессора для CNTK составляет 3.0. (Изменить: Тот факт, что вы можете запустить руководство с помощью cntk.exe, указывает на ошибку где-то в исполняемом файле v1.). Когда вы запускаете учебник с помощью cntk.exe, он распечатывает информацию о графическом процессоре, но по-прежнему использует CPU: Model has 9 nodes. Using CPU.

Единственный способ решить эту проблему - изменить значение константы MininumCCMajorForGpu в BestGpu.cpp и перекомпилируйте.

person raaar    schedule 09.02.2017
comment
Спасибо! Но согласно документ cuda takeit вычислительные возможности представляют собой поддержку различных функций. Если исходный файл был изменен, cntk будет работать без ошибок? - person freshomer; 10.02.2017
comment
Вы, конечно, можете попробовать и посмотреть, сработает ли это для вас, но все ставки, по сути, отключены. - person raaar; 10.02.2017