Google Colab GPU 加速适用于 2.x，但不适用于 1.x

Question

在https://colab.research.google.com/notebooks/gpu.ipynb 中，我认为这是 Google 对 GPU 加速的官方演示，如果我按照这些步骤进行操作，使用 Tensorflow 2.x 的 GPU 加速（比 CPU 快大约 60 倍）有效。但是，如果我想像 https://colab.research.google.com/drive/12dduH7y0GPztxSM0AFlfpjj8FU5x8YSv 那样使用 1.15 版（与第一个链接中的笔记本相比，唯一的变化是两次都去掉了“%tensorflow_version 2.x”），tf.test.gpu_device_name() 返回字符串 /device:GPU:0但没有加速。不过，我真的很想使用 1.5 和 1.15 之间的 Tensorflow 版本，因为我要运行的代码使用了在 Tensorflow 2.x 中删除的函数。有谁知道如何使用 Tensorflow 1.x，同时还能获得 GPU 加速？

Answer 1

在您的笔记本中，您的代码实际上并未执行，因为您没有调用 session.run() 或 tf.enable_eager_execution()。

在代码顶部添加tf.enable_eager_execution()，您将看到cpu 和gpu 时间之间的真正区别。