我在 Google Cloud AI Platform 上使用支持 GPU 的 Jupyter 笔记本。
然而，尽管启用了 GPU，但它只在我训练机器学习模型时使用 CPU。
有谁知道我怎样才能让它使用 GPU 来代替？在此之前我一直在使用 Google Colab 笔记本，您可以在笔记本顶部的菜单中简单地更改运行时以使用 GPU。但在 Jupyter notebook 中似乎没有这样的选项。
感谢任何关于如何让它使用 GPU 的建议！
谢谢。
【问题讨论】：
AI Platform Notebooks 没有预装驱动程序。当您create an AI Platform Notebooks instance 时，如果您选择包含 GPU，则必须选择为我自动安装 NVIDIA GPU 驱动程序选项，以便为映像提供基于框架的 CUDA 版本的最新稳定驱动程序。
【讨论】：
根据 GCP official documentation，您可以在 Jupyter Notebook 上附加 GPU。首先，您需要停止笔记本实例，然后从 GPU 列中选择 GPU 以及加速器的数量。在 AI Platform Notebooks 完成为您的笔记本实例更新 GPU 后，您可以通过选择您的实例然后单击启动来启动该实例。
【讨论】：