CPU 與 GPU 在 Keras 中的使用情況（Tensorflow 2.1）

Question

我已經成功設置了TensorFlow 2.1.0 ，可以訪問我的 GPU：

如果我使用 Keras（ from tensorflow import keras ）來適應一些Sequential model（就像這里的例子），默認情況下會使用 GPU 還是 CPU？ 是否有一些命令可以查看 Keras 正在使用哪個，我可以自己設置嗎？ 我真的很想看到一些非常基本的 Keras model 在 GPU 與 CPU 上進行訓練，以便更好地了解性能差異。

Answer 1

由於 TensorFlow 2.1、GPU 和 CPU 包都在同一個tensorflow中，不像以前的版本那樣有單獨的 CPU 和 GPU 版本： tensorflow和tensorflow-gpu 。

您可以通過這種方式測試以獲得更好的感覺：

   #Use only CPU
   import os 
   os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

或者你可以讓你的視頻卡對 TensorFlow 可見，方法是像上面一樣允許默認配置，或者通過以下方式強制它：

os.environ['CUDA_VISIBLE_DEVICES'] = '0'

請注意，在上面的設置中，如果您有 4 個 GPU，您將設置：

 os.environ['CUDA_VISIBLE_DEVICES'] = '0,1,2,3'