File tree Expand file tree Collapse file tree 1 file changed +15
-4
lines changed
doc/api/training/smd_model_parallel_release_notes Expand file tree Collapse file tree 1 file changed +15
-4
lines changed Original file line number Diff line number Diff line change @@ -49,21 +49,32 @@ SageMaker Distributed Model Parallel 1.15.0 Release Notes
4949
5050This 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
You can’t perform that action at this time.
0 commit comments