概览


Compute Engine 提供了可添加到虚拟机的图形处理单元 (GPU)。您可以使用这些 GPU 加速虚拟机上的特定工作负载,例如机器学习和数据处理。

本文档简要介绍了创建挂接 GPU 的虚拟机所需的步骤。

选择 GPU 模型

如需可用的 GPU 型号列表,请参阅 GPU 平台。记下所选 GPU 型号支持的机器类型。

对于每个模型,查看以下内容可能也会有所帮助:

选择操作系统

如果您要将 GPU 用于机器学习,则可以为虚拟机使用 Deep Learning VM Image 映像。每个 Deep Learning VM Image 映像都预安装了 GPU 驱动程序,并包含 TensorFlow 和 PyTorch 等软件包。您还可以将 Deep Learning VM Image 映像用于一般 GPU 工作负载。如需了解可用映像以及这些映像中安装的软件包,请参阅选择映像。 您还可以使用任何公共映像自定义映像,但某些映像可能需要本文档未介绍的独特驱动程序或安装过程。

您必须确定哪些驱动程序适合您的操作系统映像。如需了解安装驱动程序的步骤,请参阅安装 GPU 驱动程序

检查 GPU 配额

为了保护 Compute Engine 系统和用户,新项目会有一个 GPU 全球配额,该配额限制了您可在任何受支持可用区中创建的 GPU 总数。 如需查看 GPU 配额,请参阅 GPU 配额

如果您需要额外的 GPU 配额,请申请增加配额。申请 GPU 配额时,您必须为要在每个区域中创建的 GPU 类型申请配额,并为所有可用区中所有类型的 GPU 总数另外申请一个全球配额。

如果您的项目已建立账单历史记录,那么在您提交申请后,项目将自动获得配额。

创建挂接 GPU 的虚拟机

如需创建挂接 GPU 的虚拟机,请完成以下步骤:

  1. 创建虚拟机。用于创建虚拟机的方法取决于所选的 GPU 型号。

  2. 如需让虚拟机使用 GPU,您需要在虚拟机上安装 GPU 驱动程序。如果您启用了 NVIDIA RTX 虚拟工作站(以前称为 NVIDIA GRID),请为虚拟工作站安装驱动程序

后续步骤