2017-06-15 4 views
0

La commande "ctx = mx.cpu()" prend tout le CPU disponible. Comment restreindre pour utiliser un certain nombre seulement - disons 6 sur 8 noyauIn Deep Learning "mxnet", restreindre le nombre de cœurs (cpu)

+0

Je ne pense pas qu'il existe une telle interface. Pourquoi avez-vous réellement besoin de cela? Quoi qu'il en soit, [regardez ceci] (https://stackoverflow.com/questions/43569460/how-to-limit-pythons-use-of-cores-globally-in-ubuntu) – Mohammad

+0

A partir de maintenant, les algorithmes d'apprentissage profond dans " mxnet "prend tous les CORES disponibles et nécessite donc le contrôle. – Shiv

Répondre

1

Malheureusement - pas. Même si le contexte cpu a int comme input argument:

def cpu(device_id=0): 
    """Returns a CPU context. 

selon la documentation officielle:

Parameters 
---------- 
device_id : int, optional 
    The device id of the device. `device_id` is not needed for CPU. 
    This is included to make interface compatible with GPU. 

Cependant, en théorie, il pourrait être changé à l'avenir puisque l'argument device_id est là. Mais pour l'instant, MXNet prend tous les cœurs disponibles.