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

Posted 2023-03-16

【中文标题】为啥我的 colab 笔记本不使用 GPU？

【英文标题】：Why isn't my colab notebook using the GPU?为什么我的 colab 笔记本不使用 GPU？

【发布时间】：2020-11-01 06:38:03

【问题描述】：

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

【参考方案1】：

Colab 不仅限于 Tensorflow。

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

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

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