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

0

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

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

profile picture
EXPERTE
gefragt vor einem Jahr19 Aufrufe
1 Antwort
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
EXPERTE
beantwortet vor einem Jahr

Du bist nicht angemeldet. Anmelden um eine Antwort zu veröffentlichen.

Eine gute Antwort beantwortet die Frage klar, gibt konstruktives Feedback und fördert die berufliche Weiterentwicklung des Fragenstellers.

Richtlinien für die Beantwortung von Fragen