使用 Tensorflow2 的默认数据 + 张量位置

Question

默认情况下，使用 Tensorflow 2 时，张量以及 Keras 数据集（一旦加载）是否放置在 GPU 上？还是 CPU？

Answer 1

如果您指的是tf.data，它基本上用于RAM加载数据，并且基于tf.data管道，它同时使用CPU和GPU。 下图将让您清楚地了解这一点。