Slide 23
Slide 23 text
© 2022, Amazon Web Services, Inc. or its Affiliates.
SageMaker でサポートする DL/ML コンテナ⼀覧
• 未対応バージョンについてもコン
テナ持ち込みでご利⽤いただけま
す
※ 2022年1⽉25⽇時点
Frameworks SageMaker container supported version
Deep
Learning
TensorFlow Script mode:
2.0, 2.1, 2.2, 2.3, 2.4, 2.5, 2.6
EIA, Graviton, Neuron対応
PyTorch 0.4.0, 1.0.0, 1.1.0, 1.2.0, 1.3.0 1.4.0 1.5, 1.6, 1.7.1, 1.8.0,
1.8.1, 1.9.1
EIA, Graviton, Neuron 対応
Hugging Face TensorFlow 2.3, 2.4, 2.5, PyTorch 1.7-1.10
MXNet 0.12.1, 1.0.0, 1.1.0, 1.2.1, 1.3.0, 1.4.0, 1.4.1, 1.6.0
1.3.0, 1.4.0, 1.4.1, … 1.6, 1.7, 1.8, (for Elastic Inference)
EIA, Graviton, Neuron 対応
ML scikit-learn 0.23.1
https://github.com/aws/sagemaker-python-sdk
TensorFlow:: https://github.com/aws/sagemaker-tensorflow-serving-container
PyTorch: https://sagemaker.readthedocs.io/en/stable/using_pytorch.html
MXNet: https://sagemaker.readthedocs.io/en/stable/using_mxnet.html
Sklearn: https://sagemaker.readthedocs.io/en/stable/using_sklearn.html
Hugging Face: https://github.com/aws/deep-learning-containers/tree/master/huggingface