Skip to content

Commit e33374e

Browse files
committed
update container info
1 parent 1dd17c0 commit e33374e

File tree

1 file changed

+15
-4
lines changed

1 file changed

+15
-4
lines changed

doc/api/training/smd_model_parallel_release_notes/smd_model_parallel_change_log.rst

+15-4
Original file line numberDiff line numberDiff line change
@@ -49,21 +49,32 @@ SageMaker Distributed Model Parallel 1.15.0 Release Notes
4949

5050
This version passed benchmark testing and is migrated to the following AWS Deep Learning Containers (DLC):
5151

52-
- SageMaker training container for PyTorch v1.13.1
52+
- SageMaker training container for PyTorch v2.0.0
5353

5454
.. code::
5555
56-
763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:1.13.1-gpu-py39-cu117-ubuntu20.04-sagemaker
56+
763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.0.0-gpu-py310-cu118-ubuntu20.04-sagemaker
5757
58+
- SageMaker training container for PyTorch v1.13.1
59+
60+
.. code::
61+
62+
763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.13.1-gpu-py39-cu117-ubuntu20.04-sagemaker
5863
5964
Binary file of this version of the library for `custom container
6065
<https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-sm-sdk.html#model-parallel-bring-your-own-container>`_ users:
6166

62-
- For PyTorch 1.13.1
67+
- For PyTorch v2.0.0
6368

6469
.. code::
70+
71+
https://sagemaker-distributed-model-parallel.s3.us-west-2.amazonaws.com/pytorch-2.0.0/build-artifacts/2023-04-14-20-14/smdistributed_modelparallel-1.15.0-cp310-cp310-linux_x86_64.whl
6572
66-
https://sagemaker-distributed-model-parallel.s3.us-west-2.amazonaws.com/pytorch-1.13.1/build-artifacts/2023-01-19-18-35/smdistributed_modelparallel-1.14.0-cp39-cp39-linux_x86_64.whl
73+
- For PyTorch v1.13.1
74+
75+
.. code::
76+
77+
https://sagemaker-distributed-model-parallel.s3.us-west-2.amazonaws.com/pytorch-1.13.1/build-artifacts/2023-04-17-15-49/smdistributed_modelparallel-1.15.0-cp39-cp39-linux_x86_64.whl
6778
6879
----
6980

0 commit comments

Comments
 (0)