翻訳は機械翻訳により提供されています。提供された翻訳内容と英語版の間で齟齬、不一致または矛盾がある場合、英語版が優先します。
SageMaker モデル並列処理ライブラリ v2 のコア機能
Amazon SageMaker モデル並列処理ライブラリ v2 (SMP v2) には、分割データ並列処理、テンソル並列処理、チェックポインティングなどの分散戦略とメモリ節約技術が用意されています。SMP v2 が提供するモデル並列処理戦略と手法は、トレーニング速度とメモリ消費量を最適化しながら、大規模なモデルを複数のデバイスに分散させるのに役立ちます。SMP torch.sagemaker v2にはPythonパッケージも用意されており、数行のコード変更でトレーニングスクリプトを調整できます。
このガイドは、で紹介した基本的な 2 段階のフローに従っています。 SageMaker モデル並列処理ライブラリ v2 を使い始めるSMP v2 のコア機能とその使用方法について詳しくは、以下のトピックを参照してください。
注記
これらのコア機能は SMP v2.0.0 以降と SageMaker Python SDK v2.200.0 以降で利用でき、v2.0.1 以降でも動作します。 PyTorch パッケージのバージョンを確認するには、を参照してください。サポートされるフレームワークと AWS リージョン
トピック
