如何将 GPU 附加到 Google 云构建?

How do I attach a GPU to a Google Cloud build?

我正尝试在我的代码库上 运行 CI/CD,但为了 运行 我的测试,我需要一个支持 GPU 的虚拟机(以产生深度学习结果) .

但是,我看到的唯一可配置的机器选项是机器类型(核心数和内存)。我没有看到添加加速器类型 (GPU) 的选项。

有没有办法将 GPU 附加到构建 VM,如果没有,是否有另一种方法可以在另一个启用 GPU 的 VM 上触发测试?

谢谢!

Google Cloud Build 目前不提供配备 GPU 的机器类型。不过,一种选择是使用 remote-builder cloud builder. It allows you to run your builds on a Compute Engine instances running in your project. You can use the INSTANCE_ARGS option to customize the instance to fit your specific needs, adding one or more GPUs in this case. You can have a look here for some example configs. You can use any flag available with the gcloud compute instances create command,包括 GPU 的 --accelerator 标志。