File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change @@ -34,11 +34,8 @@ across multiple GPUs with minimal code changes. The library's API can be accesse
34
34
35
35
.. important ::
36
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
37
+ Make sure you use the pre-built Deep Learning Containers.
38
+ If you want to extend or customize your own training image,
42
39
you must use a CUDA 11 base image. For more information, see `Extend a Prebuilt Docker
43
40
Container that Contains SageMaker's Distributed Model Parallel Library
44
41
<https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-sm-sdk.html#model-parallel-customize-container> `_
You can’t perform that action at this time.
0 commit comments