Comme nous le savons, il y a un concept à propos de multiplegpu dans mxnet, quand spécifiez -gpu dans la commande. Et si nous ne spécifions pas gpu, il l'exécutera sur cpu.Comment beaucoup de processeurs le font Est-il possible de spécifier plusieurs cpu?y a-t-il un concept à propos de multiplecpu dans mxnet?
2
A
Répondre
1
Vous pouvez utiliser plusieurs processeurs avec le code suivant (version R, en python est assez similaire):
devices = lapply(1:2, function(i) {
mx.cpu(i)
})
Et le train du réseau comme d'habitude. Aussi, si vous avez MKL bibliothèque le système calcule automatiquement avec tous les cœurs.
0
Quelques façons de voir cela.
Si vous compilez MXNet avec une bonne bibliothèque BLAS, ces opérations mathématiques utiliseront tous les cœurs de processeur disponibles.
De plus, vous pouvez spécifier le nombre de threads de l'unité centrale par l'intermédiaire de la variable d'environnement MXNET_CPU_WORKER_NTHREADS
. Voir http://mxnet.io/how_to/env_var.html
Comment définir MXNET_CPU_WORKER_NTHREADS dans R – Shiv