it-swarm-es.tech

¿Cómo deshabilitar GPU en keras con tensorflow?

Quiero comparar el tiempo de procesamiento de mi código con y sin gpu. Mi backend de keras es Tensorflow. Entonces usa una GPU automáticamente. Yo uso un modelo de keras/examples/mnist_mlp.py para comparar.

Verifiqué el tiempo de procesamiento como a continuación. Entonces, ¿cómo desactivo mi GPU? Debería ~/.keras/keras.json ¿ser modificado?

$ time python mnist_mlp.py 
Test loss: 0.109761892007
Test accuracy: 0.9832
python mnist_mlp.py  38.22s user 3.18s system 162% cpu 25.543 total
8
jef

¿Has intentado algo como ésto? :

$ CUDA_VISIBLE_DEVICES='' time python mnist_mlp.py 

CUDA_VISIBLE_DEVICES generalmente se usa para ocultar algunas GPU para cuda. Aquí los oculta a todos ya que no coloca ningún dispositivo visible.

12
Nassim Ben
$ CUDA_VISIBLE_DEVICES=-1 time python mnist_mlp.py

parece ser a) ya sea la nueva forma, o b) una forma que funciona tanto en Windows como en Linux.

3
Jonas Byström