deep-learning - AWS Batch，如何申请 GPU

Question

我正在获取有关如何使用 AWS Batch 训练深度学习模型的文档。这个想法是，一旦建立了模型，我想提交几个作业来探索超参数空间。

在此interesting blog post ，博主创建了一个P2实例的执行环境，并用它来为MNIST训练一个卷积神经网络。我现在想知道是否可以在我的作业定义中要求特定数量的 GPU 而不是 vCPU。通过这种方式，我可以确定我的工作拥有所需的 GPU 数量。如果没有，是否有任何解决方法？

Answer 1

自 2019 年 4 月起，AWS Batch 开始支持 GPU 分配/调度。借助这项新功能，您可以指定作业所需的 GPU 数量。 Batch 还为您的作业进行 GPU 固定。如果一个实例有多个 GPU，Batch 可以在同一个实例上放置多个作业(每个作业需要 1 个 GPU)并让它们同时运行。下面是一个使用批处理 gpu 支持运行 gpu 作业的示例。 https://aws.amazon.com/blogs/compute/gpu-workloads-on-aws-batch/