update pytorch default version to 1.1.0

Author: Chuyang Deng

src/sagemaker/pytorch/README.rst

@@ -4,7 +4,7 @@ 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``, ``1.0.0``.
+Supported versions of PyTorch: ``0.4.0``, ``1.0.0``, ``1.1.0``.
 
 We recommend that you use the latest supported version, because that's where we focus most of our development efforts.
 
@@ -37,41 +37,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 | pytorch 1.0.0     |
-+-----------------------------+---------------+-------------------+
-| boto3                       | >=1.7.35      | >=1.9.11          |
-+-----------------------------+---------------+-------------------+
-| botocore                    | >=1.10.35     | >=1.12.11         |
-+-----------------------------+---------------+-------------------+
-| CUDA (GPU image only)       | 9.0           | 9.0               |
-+-----------------------------+---------------+-------------------+
-| 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                       | 0.4.0         | 1.0.0             |
-+-----------------------------+---------------+-------------------+
-| torchvision                 | 0.2.1         | 0.2.1             |
-+-----------------------------+---------------+-------------------+
-| Python                      | 2.7 or 3.5    | 2.7 or 3.6        |
-+-----------------------------+---------------+-------------------+
++-----------------------------+---------------+-------------------+-------------------+
+| Dependencies                | pytorch 0.4.0 | pytorch 1.0.0     | pytorch 1.1.0     |
++-----------------------------+---------------+-------------------+-------------------+
+| boto3                       | >=1.7.35      | >=1.9.11          | >=1.9.11          |
++-----------------------------+---------------+-------------------+-------------------+
+| botocore                    | >=1.10.35     | >=1.12.11         | >=1.12.11         |
++-----------------------------+---------------+-------------------+-------------------+
+| CUDA (GPU image only)       | 9.0           | 9.0               | 9.0               |
++-----------------------------+---------------+-------------------+-------------------+
+| numpy                       | >=1.14.3      | >=1.15.2          | >=1.15.2          |
++-----------------------------+---------------+-------------------+-------------------+
+| Pillow                      | >=5.1.0       | >=5.2.0           | >=5.2.0           |
++-----------------------------+---------------+-------------------+-------------------+
+| pip                         | >=10.0.1      | >=18.0            | >=18.0            |
++-----------------------------+---------------+-------------------+-------------------+
+| python-dateutil             | >=2.7.3       | >=2.7.3           | >=2.7.3           |
++-----------------------------+---------------+-------------------+-------------------+
+| retrying                    | >=1.3.3       | >=1.3.3           | >=1.3.3           |
++-----------------------------+---------------+-------------------+-------------------+
+| s3transfer                  | >=0.1.13      | >=0.1.13          | >=0.1.13          |
++-----------------------------+---------------+-------------------+-------------------+
+| sagemaker-containers        | >=2.1.0       | >=2.1.0           | >=2.1.0           |
++-----------------------------+---------------+-------------------+-------------------+
+| sagemaker-pytorch-container | 1.0           | 1.0               | 1.0               |
++-----------------------------+---------------+-------------------+-------------------+
+| setuptools                  | >=39.2.0      | >=40.4.3          | >=40.4.3          |
++-----------------------------+---------------+-------------------+-------------------+
+| six                         | >=1.11.0      | >=1.11.0          | >=1.11.0          |
++-----------------------------+---------------+-------------------+-------------------+
+| torch                       | 0.4.0         | 1.0.0             | 1.0.0             |
++-----------------------------+---------------+-------------------+-------------------+
+| torchvision                 | 0.2.1         | 1.1.0             | 0.3.0             |
++-----------------------------+---------------+-------------------+-------------------+
+| Python                      | 2.7 or 3.5    | 2.7 or 3.6        | 2.7 or 3.6        |
++-----------------------------+---------------+-------------------+-------------------+
 
 The Docker images extend Ubuntu 16.04.
 

src/sagemaker/pytorch/defaults.py

@@ -12,7 +12,7 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
-PYTORCH_VERSION = '0.4'
+PYTORCH_VERSION = '1.1.0'
 """Default PyTorch version for when the framework version is not specified.
 This is no longer updated so as to not break existing workflows.
 """

src/sagemaker/pytorch/estimator.py

@@ -29,7 +29,7 @@ class PyTorch(Framework):
 
     __framework_name__ = "pytorch"
 
-    LATEST_VERSION = '1.0'
+    LATEST_VERSION = '1.1.0'
     """The latest version of PyTorch included in the SageMaker pre-built Docker images."""
 
     def __init__(self, entry_point, source_dir=None, hyperparameters=None, py_version=PYTHON_VERSION,