如何将 GPU 连接到 Google Cloud 构建？

Question

我正在尝试在我的代码库上运行 CI/CD，但为了运行我的测试，我需要一个支持 GPU 的虚拟机（以生成深度学习结果）。

然而，我看到的唯一可配置的机器选项是机器类型（内核和内存的数量）。我没有看到添加加速器类型 (GPU) 的选项。

有没有办法将 GPU 连接到构建虚拟机，如果没有，是否有另一种方法可以在另一个启用 GPU 的虚拟机上触发测试？

谢谢！

Answer 1

Google Cloud Build 目前不提供配备 GPU 的机器类型。一种选择是使用远程构建器云构建器。它允许您在项目中运行的 Compute Engine 实例上运行构建。您可以使用该INSTANCE_ARGS选项自定义实例以满足您的特定需求，在这种情况下添加一个或多个 GPU。您可以在此处查看一些示例配置。gcloud compute instances create 您可以使用该命令可用的任何标志，包括--acceleratorGPU 的标志。