テンソルフローでケラスで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に隠すために使用されます。ここでは、目に見えるデバイスを置かないので、それらをすべて非表示にします。