it-swarm.com.de

Wie deaktiviere ich die GPU in Keras mit Tensorflow?

Ich möchte die Verarbeitungszeit meines Codes mit und ohne GPU vergleichen. Mein Backend von Keras ist Tensorflow. Es wird also automatisch eine GPU verwendet. Ich benutze ein Modell von keras/examples/mnist_mlp.py zum Vergleichen.

Ich habe die Bearbeitungszeit wie unten angegeben überprüft. Wie deaktiviere ich dann meine GPU? Sollte ~/.keras/keras.json geändert werden?

$ 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
7
jef

Hast du so etwas probiert? :

$ CUDA_VISIBLE_DEVICES='' time python mnist_mlp.py 

CUDA_VISIBLE_DEVICES wird normalerweise verwendet, um einige GPUs in cuda zu verbergen. Hier verstecken Sie sie alle, da Sie kein sichtbares Gerät ablegen. 

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

scheint a) entweder der neue Weg oder b) ein Weg, der sowohl unter Windows als auch unter Linux funktioniert.

0
Jonas Byström