Instruction on how to use preview version of PyTorch - 1.0.0.dev. (#407)

Author: nadiaya

README.rst

@@ -286,10 +286,13 @@ PyTorch SageMaker Estimators
 
 With PyTorch SageMaker ``Estimators``, you can train and host PyTorch models on Amazon SageMaker.
 
-Supported versions of PyTorch: ``0.4.0``.
+Supported versions of PyTorch: ``0.4.0``, ``1.0.0.dev`` ("Preview").
 
 We recommend that you use the latest supported version, because that's where we focus most of our development efforts.
 
+You can try the "Preview" version of PyTorch by specifying ``'1.0.0.dev'`` for ``framework_version`` when creating your PyTorch estimator.
+This will ensure you're using the latest version of ``torch-nightly``.
+
 For more information about PyTorch, see https://github.com/pytorch/pytorch.
 
 For more information about PyTorch SageMaker ``Estimators``, see `PyTorch SageMaker Estimators and Models`_.

src/sagemaker/pytorch/README.rst

@@ -4,7 +4,10 @@ SageMaker PyTorch Estimators and Models
 
 With PyTorch Estimators and Models, you can train and host PyTorch models on Amazon SageMaker.
 
-Supported versions of PyTorch: ``0.4.0``.
+Supported versions of PyTorch: ``0.4.0``, ``1.0.0.dev`` ("Preview").
+
+You can try the "Preview" version of PyTorch by specifying ``1.0.0.dev`` for ``framework_version`` when creating your PyTorch estimator.
+This will ensure you're using the latest version of ``torch-nightly``.
 
 You can visit the PyTorch repository at https://github.com/pytorch/pytorch.
 
@@ -660,41 +663,41 @@ This Python version applies to both the Training Job, created by fit, and the En
 
 The PyTorch Docker images have the following dependencies installed:
 
-+-----------------------------+---------------+
-| Dependencies                | pytorch 0.4.0 |
-+-----------------------------+---------------+
-| boto3                       | 1.7.35        |
-+-----------------------------+---------------+
-| botocore                    | 1.10.35       |
-+-----------------------------+---------------+
-| CUDA (GPU image only)       | 9.0           |
-+-----------------------------+---------------+
-| numpy                       | 1.14.3        |
-+-----------------------------+---------------+
-| Pillow                      | 5.1.0         |
-+-----------------------------+---------------+
-| pip                         | 10.0.1        |
-+-----------------------------+---------------+
-| python-dateutil             | 2.7.3         |
-+-----------------------------+---------------+
-| retrying                    | 1.3.3         |
-+-----------------------------+---------------+
-| s3transfer                  | 0.1.13        |
-+-----------------------------+---------------+
-| sagemaker-containers        | 2.0.4         |
-+-----------------------------+---------------+
-| sagemaker-pytorch-container | 1.0           |
-+-----------------------------+---------------+
-| setuptools                  | 39.2.0        |
-+-----------------------------+---------------+
-| six                         | 1.11.0        |
-+-----------------------------+---------------+
-| torch                       | 0.4.0         |
-+-----------------------------+---------------+
-| torchvision                 | 0.2.1         |
-+-----------------------------+---------------+
-| Python                      | 2.7 or 3.5    |
-+-----------------------------+---------------+
++-----------------------------+---------------+-------------------+
+| Dependencies                | pytorch 0.4.0 | pytorch 1.0.0.dev |
++-----------------------------+---------------+-------------------+
+| boto3                       | >=1.7.35      | >=1.9.11          |
++-----------------------------+---------------+-------------------+
+| botocore                    | >=1.10.35     | >=1.12.11         |
++-----------------------------+---------------+-------------------+
+| CUDA (GPU image only)       | 9.0           | 9.2               |
++-----------------------------+---------------+-------------------+
+| numpy                       | >=1.14.3      | >=1.15.2          |
++-----------------------------+---------------+-------------------+
+| Pillow                      | >=5.1.0       | >=5.2.0           |
++-----------------------------+---------------+-------------------+
+| pip                         | >=10.0.1      | >=18.0            |
++-----------------------------+---------------+-------------------+
+| python-dateutil             | >=2.7.3       | >=2.7.3           |
++-----------------------------+---------------+-------------------+
+| retrying                    | >=1.3.3       | >=1.3.3           |
++-----------------------------+---------------+-------------------+
+| s3transfer                  | >=0.1.13      | >=0.1.13          |
++-----------------------------+---------------+-------------------+
+| sagemaker-containers        | >=2.1.0       | >=2.1.0           |
++-----------------------------+---------------+-------------------+
+| sagemaker-pytorch-container | 1.0           | 1.0               |
++-----------------------------+---------------+-------------------+
+| setuptools                  | >=39.2.0      | >=40.4.3          |
++-----------------------------+---------------+-------------------+
+| six                         | >=1.11.0      | >=1.11.0          |
++-----------------------------+---------------+-------------------+
+| torch (torch-nightly)       | 0.4.0         | 1.0.0.dev         |
++-----------------------------+---------------+-------------------+
+| torchvision                 | 0.2.1         | 0.2.1             |
++-----------------------------+---------------+-------------------+
+| Python                      | 2.7 or 3.5    | 2.7 or 3.5        |
++-----------------------------+---------------+-------------------+
 
 The Docker images extend Ubuntu 16.04.