@@ -11,10 +11,11 @@ across multiple GPUs with minimal code changes. The library's API can be accesse
11
11
12
12
.. tip ::
13
13
14
- We recommended using this API documentation with the conceptual guide at
14
+ We recommend that you use this API documentation along with the conceptual guide at
15
15
`SageMaker's Distributed Model Parallel
16
16
<http://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel.html> `_
17
- in the *Amazon SageMaker developer guide *. This developer guide documentation includes:
17
+ in the *Amazon SageMaker developer guide *.
18
+ The conceptual guide includes the following topics:
18
19
19
20
- An overview of model parallelism, and the library's
20
21
`core features <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-core-features.html >`_,
@@ -32,10 +33,14 @@ across multiple GPUs with minimal code changes. The library's API can be accesse
32
33
33
34
34
35
.. important ::
35
- The model parallel library only supports training jobs using CUDA 11. When you define a PyTorch or TensorFlow
36
- ``Estimator `` with ``modelparallel `` parameter ``enabled `` set to ``True ``,
37
- it uses CUDA 11. When you extend or customize your own training image
38
- you must use a CUDA 11 base image. See
39
- `Extend or Adapt A Docker Container that Contains the Model Parallel Library
40
- <https://integ-docs-aws.amazon.com/sagemaker/latest/dg/model-parallel-use-api.html#model-parallel-customize-container> `__
41
- for more information.
36
+ The model parallel library only supports SageMaker training jobs using CUDA 11.
37
+ Make sure you use the pre-built Deep Learning Containers, or use the right CUDA version
38
+ if you use a custom training container.
39
+
40
+ .. tip ::
41
+ If you want to extend or customize your own training image
42
+ you must use a CUDA 11 base image. For more information, see `Extend a Prebuilt Docker
43
+ Container that Contains SageMaker's Distributed Model Parallel Library
44
+ <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-sm-sdk.html#model-parallel-customize-container> `_
45
+ and `Create Your Own Docker Container with the SageMaker Distributed Model Parallel Library
46
+ <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-sm-sdk.html#model-parallel-bring-your-own-container> `_.
0 commit comments