如何在具有张量流的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。在这里你隐藏它们，因为你没有放任何可见的设备。

Answer 2

$ CUDA_VISIBLE_DEVICES=-1 time python mnist_mlp.py

似乎是a）新方式，或b）适用于Windows和Linux的方式。