Comment désactiver le GPU dans les keras avec tensorflow?
Je veux comparer le temps de traitement de mon code avec et sans gpu. Mon backend de keras est Tensorflow. Donc, il utilise un GPU automatiquement. J'utilise un modèle de keras/examples/mnist_mlp.py pour comparer.
J'ai vérifié le temps de traitement comme ci-dessous. Alors, comment puis-je désactiver mon GPU? ~/.keras/keras.json devrait-il être modifié?
$ 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
Avez-vous essayé quelque chose comme ça? :
$ CUDA_VISIBLE_DEVICES='' time python mnist_mlp.py
CUDA_VISIBLE_DEVICES est généralement utilisé pour masquer certains GPU à cuda. Ici, vous les cachez tous car vous ne mettez aucun périphérique visible.
$ CUDA_VISIBLE_DEVICES=-1 time python mnist_mlp.py
semble être a) soit la nouvelle manière, ou b) une manière qui fonctionne à la fois sous Windows et Linux.