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'