Qual è il formato dei filtri del dispositivo in TensorFlow?

Question

Così il proto sessione config ha un'opzione device_filters, con il commento:

// When any filters are present sessions will ignore all devices which do not 
    // match the filters. Each filter can be partially specified, e.g. "/job:ps" 
    // "/job:worker/replica:3", etc. 

Qualcuno ha una spiegazione concreta del formato? Ad esempio, voglio escludere/gpu: 0 come opzione perché lo uso per far girare altri modelli.

Ho provato

config = tf.ConfigProto() 
config.device_filters.append('/gpu:1') 
config.device_filters.append('/cpu:0') 
with tf.Session(config=config): 
    # Do stuff 

Ma sto ancora ricevendo ops assegnati alla gpu 0. non voglio ignorare i dispositivi su una base per-op.

Answer 1

Il campo ConfigProto.device_filters è attualmente ignorato da TensorFlow, anche se è destinato a supportare il caso d'uso in futuro. Se si vuole raggiungere lo stesso fine ops in esecuzione su /gpu:1 e /cpu:0, si può fare come segue, l'utilizzo di "placement soft":

with tf.device("/gpu:1"): 
    # Build your model in this with context. All nodes will get the 
    # device "/gpu:1". 

with tf.Session(config=tf.ConfigProto(allow_soft_placement=True)): 
    # Execute your mode in this with context. 
    # Soft placement will use /gpu:1 for GPU-compatible ops, and /cpu:0 
    # for CPU-only ops.