Tensorflow: esecuzione di un'operazione con un core specifico di una CPU

Question

Attualmente è possibile specificare quale CPU o GPU utilizzare con la funzione tf.device (...) per operazioni specifiche, ma è comunque possibile specificare a core di una CPU?

Answer 1

Al momento non ci sono API per bloccare le operazioni su un particolare core, sebbene ciò renderebbe un buon feature request. Si potrebbe approssimare questa funzionalità creando più dispositivi CPU, ciascuno con un singolo thread ThreadPool, ma questo non è garantito per mantenere la posizione di una soluzione core-pinning:

with tf.device("/cpu:4"): 
    # ... 

with tf.device("/cpu:7"): 
    # ... 

with tf.device("/cpu:0"): 
    # ... 

config = tf.ConfigProto(device_count={"CPU": 8}, 
         inter_op_parallelism_threads=1, 
         intra_op_parallelism_threads=1) 
sess = tf.Session(config=config)