Amazon SageMaker で PyTorch を使用 - Amazon SageMaker

Amazon SageMaker で PyTorch を使用

カスタム PyTorch コードを使ってモデルをトレーニングしデプロイするために Amazon SageMaker を使うことができます。Amazon SageMaker Python SDK の PyTorch 推定器やモデル、および Amazon SageMaker オープンソースの PyTorch コンテナでは、Amazon SageMaker での PyTorch スクリプトの記述や実行がより簡単になります。

何をしたいですか?

Amazon SageMaker で、カスタムの PyTorch モデルをトレーニングしたいと思います。

サンプルの Jupyter ノートブックについては、https://github.com/awslabs/amazon-sagemaker-examples/tree/master/sagemaker-python-sdk/pytorch_mnist を参照してください。

ドキュメントについては、「PyTorch によるモデルのトレーニング」を参照してください。

Amazon SageMaker でトレーニングした PyTorch モデルがあり、それをホスト済みのエンドポイントにデプロイしたいと考えています。

PyTorch モデルをデプロイします

Amazon SageMaker の外部でトレーニングした PyTorch モデルがあり、それを Amazon SageMaker エンドポイントにデプロイしたいと考えています。

モデルデータからエンドポイントをデプロイします

Amazon SageMaker Python SDK PyTorch クラスの API ドキュメントを表示したいと思います。

PyTorch クラス

Amazon SageMaker PyTorch コンテナに関する情報を表示したいと思います。

https://github.com/aws/sagemaker-pytorch-container

Amazon SageMaker での PyTorch トレーニングスクリプトの記述や、PyTorch 推定器およびモデルの使用に関する一般的な情報については、「SageMaker Python SDK での PyTorch の使用」を参照してください。

Amazon SageMaker PyTorch コンテナでサポートされている PyTorch バージョンについては、「https://github.com/aws/sagemaker-python-sdk/blob/master/src/sagemaker/pytorch/README.rst」を参照してください。