2016-05-31 6 views
2

Как известно, в mxnet существует понятие о multiplegpu, когда указывается -gpu в команде. И если мы не укажем gpu, он запустит его на cpu.How многие cpus делают это запустить? Можно ли указать несколько процессоров?есть концепция о multipcpu в mxnet

ответ

1

Вы можете использовать несколько процессоров со следующим кодом (R версии, в питон очень похож):

devices = lapply(1:2, function(i) { 
    mx.cpu(i) 
}) 

И поезд сети, как обычно. Также, если у вас есть библиотека MKL, система автоматически вычисляет все ядра.

0

Пара способов взглянуть на это.

Если вы скомпилируете MXNet с хорошей библиотекой BLAS, эти математические операции будут использовать все доступные ядра процессора.

Кроме того, вы можете указать, сколько рабочих потоков ЦП через переменную окружения MXNET_CPU_WORKER_NTHREADS. См. http://mxnet.io/how_to/env_var.html

+0

Как установить MXNET_CPU_WORKER_NTHREADS в R – Shiv