哪些Amazon SageMaker算法只能使用GPU进行训练？
0
【以下的问题经过翻译处理】 我在某个资料中看到，部分Amazon SageMaker内置的算法仅支持使用GPU进行训练，而有一些算法可以使用GPU或CPU，还有一些算法只能在CPU上使用。
是否有官方文档明确说明哪些算法只能使用GPU或两者都可以使用？
1 回答
- 最新
- 投票最多
- 评论最多
0
【以下的回答经过翻译处理】 Amazon SageMaker内置算法文档中提供了关于选择Amazon EC2实例以及特定算法是否支持GPU或CPU设备的建议。
以图像分类为例，以下是在线文档的摘录：
对于图像分类，我们支持以下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模式，客户可以选择使用GPU还是CPU设备来执行不同的操作。这是作为他们的训练脚本的一部分进行配置的。