CPU 对比 Keras 中的 GPU 使用情况（Tensorflow 2.1）

Question

我已成功设置 TensorFlow 2.1.0 以访问我的 GPU：

如果我使用 Keras (from tensorflow import keras) 来拟合某些 Sequential 模型（例如 here），默认情况下将使用 GPU 或 CPU ？是否有一些命令可以查看 Keras 使用的是哪一个，我可以以某种方式自己设置它吗？我真的很想看到一些非常基本的 Keras 模型在 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'

CPU 对比 Keras 中的 GPU 使用情况（Tensorflow 2.1）

CPU vs GPU usage in Keras (Tensorflow 2.1)

python

cpu

gpu

keras

tensorflow2.0