TensorFlowを使用してケラスでGPUを無効にする方法は？

Question

コードの処理時間をgpuありとなしで比較したいと思います。私のケラスのバックエンドは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の両方で機能する方法のようです。