Compute Engine 提供了可添加到虚拟机的图形处理单元 (GPU)。您可以使用这些 GPU 加速虚拟机上的特定工作负载,例如机器学习和数据处理。
本文档简要介绍了创建挂接了 GPU 的虚拟机所需的步骤。
如需详细了解 Compute Engine 上的 GPU,请参阅 GPU 简介。
选择 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 的虚拟机,请完成以下步骤:
创建虚拟机。用于创建虚拟机的方法取决于所选的 GPU 型号。
- 如需创建挂接了 NVIDIA H100、A100 或 L4 GPU 的虚拟机,请参阅创建加速器优化虚拟机。
- 如需创建挂接了 NVIDIA T4、P4、P100 或 V100 GPU 的虚拟机,请参阅创建挂接了 GPU 的 N1 虚拟机。
如需让虚拟机使用 GPU,您需要在虚拟机上安装 GPU 驱动程序。如果您启用了 NVIDIA RTX 虚拟工作站(以前称为 NVIDIA GRID),请为虚拟工作站安装驱动程序。