（アーカイブ） SageMaker モデル並列処理ライブラリ v1.x

Amazon SageMaker AI のモデル並列ライブラリを使用して、GPUメモリ制限のためにトレーニングが困難な大規模深層学習 (DL) モデルをトレーニングします。ライブラリは、モデルを複数の および インスタンス間で自動的かつ効率的に分割GPUsします。ライブラリを使用すると、数十億または数兆のパラメータを使って大規模な DL モデルを効率的にトレーニングすることで、目標予測精度をより速く達成できます。

ライブラリを使用すると、コードの変更を最小限に抑えながら、複数のノードGPUsと複数のノードに独自の TensorFlow および PyTorch モデルを自動的にパーティション化できます。Python APIを使用してライブラリの SageMaker にアクセスできますSDK。

モデル並列処理とモデル SageMaker 並列ライブラリの詳細については、以下のセクションを参照してください。このライブラリのAPIドキュメントは、SageMaker Python v2SDK.199.0 ドキュメントの分散トレーニングAPIsにあります。