哪些Amazon SageMaker算法只能使用GPU进行训练？

Question

【以下的问题经过翻译处理】 我在某个资料中看到，部分Amazon SageMaker内置的算法*仅支持*使用GPU进行训练，而有一些算法可以使用GPU或CPU，还有一些算法只能在CPU上使用。

是否有官方文档明确说明哪些算法只能使用GPU或两者都可以使用？

Answer

【以下的回答经过翻译处理】 [Amazon SageMaker内置算法文档](https://docs.aws.amazon.com/sagemaker/latest/dg/algos.html)中提供了关于选择Amazon EC2实例以及特定算法是否支持GPU或CPU设备的建议。

以[图像分类](https://docs.aws.amazon.com/sagemaker/latest/dg/image-classification.html)为例，以下是在线文档的摘录：

> 对于图像分类，我们支持以下GPU实例用于训练：ml.p2.xlarge，ml.p2.8xlarge，ml.p2.16xlarge，ml.p3.2xlarge，ml.p3.8xlarge和ml.p3.16xlarge。我们建议使用具有更多内存的GPU实例来进行大批量训练。然而，推理可以使用CPU(例如C4)和GPU(例如P2和P3)实例。您还可以在多GPU和多实例的环境中进行分布式训练。

对于更复杂的情况，例如[脚本或BYOC](https://docs.aws.amazon.com/sagemaker/latest/dg/your-algorithms.html)模式，客户可以选择使用GPU还是CPU设备来执行不同的操作。这是作为他们的训练脚本的一部分进行配置的。

哪些Amazon SageMaker算法只能使用GPU进行训练？

Relevant content