为什么我的 colab 笔记本不使用 GPU？

Question

当我在选择 GPU 后在我的 colab 笔记本上运行代码时，我收到一条消息，提示“您已连接到 GPU 运行时，但未使用 GPU”。现在我明白以前有人问过类似的问题，但我仍然不明白为什么。我在数百次迭代的数据集上运行 PCA，进行多次试验。如果没有 GPU，它需要的时间与我的笔记本电脑一样长，可能超过 12 小时，导致 colab 超时。 colab 的 GPU 是否仅限于 tensorflow 等机器学习库？有没有办法解决这个问题，以便我可以利用 GPU 来加快分析速度？

Answer 1

Colab 不仅限于 Tensorflow。

Colab 提供三种运行时：标准运行时（带有 CPU）、GPU 运行时（包括 GPU）和 TPU 运行时（包括 TPU）。

“您已连接到 GPU 运行时，但未使用 GPU”表示用户已连接到 GPU 运行时，但未使用 GPU，因此成本较低的 CPU 运行时会更合适。

因此，您必须使用利用 GPU 的软件包，例如 Tensorflow 或 Jax。 GPU 运行时也有一个 CPU，除非您专门使用运行 GPU 的包，否则它将处于空闲状态。