Как известно, в mxnet существует понятие о multiplegpu, когда указывается -gpu в команде. И если мы не укажем gpu, он запустит его на cpu.How многие cpus делают это запустить? Можно ли указать несколько процессоров?есть концепция о multipcpu в mxnet
2
A
ответ
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
Как установить MXNET_CPU_WORKER_NTHREADS в R – Shiv