如何用tensorflow禁用keras中的GPU？

Question

我想比较我的代码处理时间和不使用gpu.我的keras后端是Tensorflow.所以它会自动使用GPU.我用一个keras/examples/mnist_mlp.py比较模型.

我检查了下面的处理时间.那么,我如何禁用我的GPU？应该~/.keras/keras.json修改？

$ 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

Answer 1

你尝试过这样的事吗？:

$ CUDA_VISIBLE_DEVICES='' time python mnist_mlp.py 

CUDA_VISIBLE_DEVICES通常用于隐藏一些GPU到cuda.在这里你隐藏它们,因为你没有放任何可见的设备.