Passaggio del tensore tra CPU/GPU

Question

Dopo aver installato la GPU di tensorflow (in esecuzione su un measly NVIDIA GeForce 950), vorrei confrontare le prestazioni con la CPU.

Sto eseguendo il codice del tutorial di tensorFlow MNIST e ho notato un notevole aumento di velocità, stimato comunque (ho eseguito la versione della CPU 2 giorni fa su un portatile i7 con una dimensione di 100, e questo su un desktop GPU, dimensione batch di 10) - tra la CPU e la GPU quando sono passato ... ma ho notato solo l'aumento di velocità quando ho ridotto la dimensione del batch sulla GPU a 10 da 100 ...

Ora I manca una misura oggettiva per quello che sto guadagnando.

C'è un modo per passare tra i flussi del tensore della CPU e della GPU?

Answer 1

Per rendere invisibile GPU

export CUDA_VISIBLE_DEVICES="" 

Per tornare alla normalità

unset CUDA_VISIBLE_DEVICES 

Answer 2

provare a impostare tf.device di CPU: 0

with tf.Session() as sess: 
    with tf.device("/cpu:0"): 

Answer 3

Un'altra opzione è quella di installare la versione della CPU e la versione gpu di tensorflow in due ambienti virtuali, istruzioni dettagliate su come installare tensorflow in ambiente virtuale le voci sono elencate qui https://www.tensorflow.org/get_started/os_setup; in questo modo, è possibile avere lo stesso codice in esecuzione in due finestre di terminali, una utilizza la CPU e l'altra utilizza la GPU.