如何使用張量鎖定keras中的GPU？

Question

我想比較我的代碼與gpu和沒有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。在這裏你可以隱藏它們，因爲你沒有放置任何可見的設備。