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

0

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

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

profile picture
专家
已提问 7 个月前12 查看次数
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设备来执行不同的操作。这是作为他们的训练脚本的一部分进行配置的。

profile picture
专家
已回答 7 个月前

您未登录。 登录 发布回答。

一个好的回答可以清楚地解答问题和提供建设性反馈,并能促进提问者的职业发展。

回答问题的准则