update pytorch page

Author: mchoi8739

doc/api/training/sdp_versions/latest.rst

@@ -15,9 +15,10 @@ Select the latest or one of the previous versions of the API documentation
 depending on the version of the library you use.
 
 .. important::
-   The distributed data parallel library only supports training jobs using CUDA 11. When you define a PyTorch or TensorFlow
-   ``Estimator`` with ``dataparallel`` parameter ``enabled`` set to ``True``,
-   it uses CUDA 11. When you extend or customize your own training image
+   The distributed data parallel library only supports training jobs using CUDA 11 or later.
+   When you define a SageMaker ``PyTorch`` or ``TensorFlow``
+   estimator with ``dataparallel`` parameter ``enabled`` set to ``True``,
+   it uses CUDA 11. When you extend or customize your own training image,
    you must use a CUDA 11 base image. See
    `SageMaker Python SDK's distributed data parallel library APIs
    <https://docs.aws.amazon.com/sagemaker/latest/dg/data-parallel-use-api.html#data-parallel-use-python-skd-api>`_
@@ -32,12 +33,12 @@ Version 1.4.0 (Latest)
    latest/smd_data_parallel_pytorch
    latest/smd_data_parallel_tensorflow
 
-To find archived API documentation for the previous versions of the library,
-see the following link:
-
 Documentation Archive
 =====================
 
+To find the API documentation for the previous versions of the library,
+choose one of the following:
+
 .. toctree::
    :maxdepth: 1