我想比较使用和不使用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
最佳答案
你尝试过这样的事情吗? :
$ CUDA_VISIBLE_DEVICES='' time python mnist_mlp.py
CUDA_VISIBLE_DEVICES
通常用于将某些GPU隐藏到cuda中。在这里,您将它们全部隐藏起来,因为您没有放置任何可见的设备。
关于tensorflow - 如何使用Tensorflow在Keras中禁用GPU?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43070568/