Skip to content

Commit a66c48d

Browse files
chuyang-dengChuyang Deng
chuyang-deng
and
Chuyang Deng
authored
feature: PyTorch 1.5.0 support (#1460)
Co-authored-by: Chuyang Deng <[email protected]>
1 parent 1d6f54f commit a66c48d

File tree

4 files changed

+47
-43
lines changed

4 files changed

+47
-43
lines changed

‎README.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -246,7 +246,7 @@ PyTorch SageMaker Estimators
246246

247247
With PyTorch SageMaker Estimators, you can train and host PyTorch models on Amazon SageMaker.
248248

249-
Supported versions of PyTorch: ``0.4.0``, ``1.0.0``, ``1.1.0``, ``1.2.0``, ``1.3.1``, ``1.4.0``.
249+
Supported versions of PyTorch: ``0.4.0``, ``1.0.0``, ``1.1.0``, ``1.2.0``, ``1.3.1``, ``1.4.0``, ``1.5.0``.
250250

251251
Supported versions of PyTorch for Elastic Inference: ``1.3.1``.
252252

‎doc/using_pytorch.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ Using PyTorch with the SageMaker Python SDK
44

55
With PyTorch Estimators and Models, you can train and host PyTorch models on Amazon SageMaker.
66

7-
Supported versions of PyTorch: ``0.4.0``, ``1.0.0``, ``1.1.0``, ``1.2.0``, ``1.3.1``, ``1.4.0``.
7+
Supported versions of PyTorch: ``0.4.0``, ``1.0.0``, ``1.1.0``, ``1.2.0``, ``1.3.1``, ``1.4.0``, ``1.5.0``.
88

99
Supported versions of PyTorch for Elastic Inference: ``1.3.1``.
1010

‎src/sagemaker/pytorch/README.rst

Lines changed: 44 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ SageMaker PyTorch Estimators and Models
44

55
With PyTorch Estimators and Models, you can train and host PyTorch models on Amazon SageMaker.
66

7-
Supported versions of PyTorch: ``0.4.0``, ``1.0.0``, ``1.1.0``, ``1.2.0``, ``1.3.1``, ``1.4.0``.
7+
Supported versions of PyTorch: ``0.4.0``, ``1.0.0``, ``1.1.0``, ``1.2.0``, ``1.3.1``, ``1.4.0``, ``1.5.0``.
88

99
Supported versions of PyTorch for Elastic Inference: ``1.3.1``.
1010

@@ -39,45 +39,49 @@ This Python version applies to both the Training Job, created by fit, and the En
3939

4040
The PyTorch Docker images have the following dependencies installed:
4141

42-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
43-
| Dependencies | pytorch 0.4.0 | pytorch 1.0.0 | pytorch 1.1.0 | pytorch 1.2.0 | pytorch 1.3.1 | pytorch 1.4.0 |
44-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
45-
| boto3 | >=1.7.35 | >=1.9.11 | 1.9.82 | 1.9.249 | 1.10.34 | 1.12.4 |
46-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
47-
| botocore | >=1.10.35 | >=1.12.11 | >= 1.12.11 | 1.12.249 | 1.13.34 | 1.15.4 |
48-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
49-
| CUDA (GPU image only) | 9.0 | 9.0 | 10.1 | 10.0 | 10.1 | 10.1 |
50-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
51-
| numpy | >=1.14.3 | >=1.15.2 | 1.16.4 | 1.16.4 | 1.16.4 | 1.16.4 |
52-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
53-
| Pillow | >=5.1.0 | >=5.2.0 | 6.0.0 | 5.4.1 | 6.2.1 | 6.2.0 |
54-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
55-
| pip | >=10.0.1 | >=18.0 | >=18.0 | 19.3 | 19.3.1 | 20.0.2 |
56-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
57-
| python-dateutil | >=2.7.3 | >=2.7.3 | >=2.7.3 | 2.8.0 | 2.8.0 | 2.8.1 |
58-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
59-
| retrying | >=1.3.3 | >=1.3.3 | 1.3.3 | 1.3.3 | 1.3.3 | 1.3.3 |
60-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
61-
| s3transfer | >=0.1.13 | >=0.1.13 | >=0.1.13 | 0.2.1 | 0.2.1 | 0.3.3 |
62-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
63-
| sagemaker-containers | >=2.1.0 | >=2.1.0 | 2.4.10.post0 | 2.5.4 | 2.5.4 | 2.5.4 |
64-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
65-
| sagemaker-inference (inference only) | N/A | N/A | N/A | N/A | 1.1.2 | 1.1.2 |
66-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
67-
| sagemaker-pytorch-container | 1.0 | 1.1 | 1.2 | 1.2 | 1.3 | N/A |
68-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
69-
| sagemaker-pytorch-inference | N/A | N/A | N/A | N/A | N/A | 1.1.2 |
70-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
71-
| setuptools | >=39.2.0 | >=40.4.3 | >=40.4.3 | 41.4.0 | 42.0.2.post20191203 | N/A |
72-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
73-
| six | >=1.11.0 | >=1.11.0 | 1.12.0 | 1.12.0 | 1.12.0 | 1.12.0 |
74-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
75-
| torch | 0.4.0 | 1.0.0 | 1.1.0 | 1.2.0 | 1.3.1 | 1.4.0 |
76-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
77-
| torchvision | 0.2.1 | 0.2.1 | 0.3.0 | 0.4.0a0+9232c4a | 0.4.2 | 0.5.0 |
78-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
79-
| Python | 2.7 or 3.5 | 2.7 or 3.6 | 2.7 or 3.6 | 2.7 or 3.6 | 2.7 or 3.6 | 2.7 or 3.6 |
80-
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+
42+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
43+
| Dependencies | pytorch 0.4.0 | pytorch 1.0.0 | pytorch 1.1.0 | pytorch 1.2.0 | pytorch 1.3.1 | pytorch 1.4.0 | pytorch 1.5.0 |
44+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
45+
| boto3 | >=1.7.35 | >=1.9.11 | 1.9.82 | 1.9.249 | 1.10.34 | 1.12.4 | 1.10.32 |
46+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
47+
| botocore | >=1.10.35 | >=1.12.11 | >= 1.12.11 | 1.12.249 | 1.13.34 | 1.15.4 | 1.16.3 |
48+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
49+
| CUDA (GPU image only) | 9.0 | 9.0 | 10.1 | 10.0 | 10.1 | 10.1 | 10.1 |
50+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
51+
| numpy | >=1.14.3 | >=1.15.2 | 1.16.4 | 1.16.4 | 1.16.4 | 1.16.4 | 1.16.4 |
52+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
53+
| Pillow | >=5.1.0 | >=5.2.0 | 6.0.0 | 5.4.1 | 6.2.1 | 6.2.0 | 7.1.0 |
54+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
55+
| pip | >=10.0.1 | >=18.0 | >=18.0 | 19.3 | 19.3.1 | 20.0.2 | 20.0.2 |
56+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
57+
| python-dateutil | >=2.7.3 | >=2.7.3 | >=2.7.3 | 2.8.0 | 2.8.0 | 2.8.1 | 2.8.1 |
58+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
59+
| retrying | >=1.3.3 | >=1.3.3 | 1.3.3 | 1.3.3 | 1.3.3 | 1.3.3 | 1.3.3 |
60+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
61+
| s3transfer | >=0.1.13 | >=0.1.13 | >=0.1.13 | 0.2.1 | 0.2.1 | 0.3.3 | 0.3.3 |
62+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
63+
| sagemaker-containers | >=2.1.0 | >=2.1.0 | 2.4.10.post0 | 2.5.4 | 2.5.4 | 2.5.4 | N/A |
64+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
65+
| sagemaker-training (training only) | N/A | N/A | N/A | N/A | N/A | N/A | >=3.4.2 |
66+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
67+
| sagemaker-inference (inference only) | N/A | N/A | N/A | N/A | 1.1.2 | 1.1.2 | >=1.2.2 |
68+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
69+
| sagemaker-pytorch-container | 1.0 | 1.1 | 1.2 | 1.2 | 1.3 | N/A | N/A |
70+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
71+
| sagemaker-pytorch-training | N/A | N/A | N/A | N/A | N/A | N/A | >=2.1.1 |
72+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
73+
| sagemaker-pytorch-inference | N/A | N/A | N/A | N/A | N/A | 1.1.2 | >=1.2.2 |
74+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
75+
| setuptools | >=39.2.0 | >=40.4.3 | >=40.4.3 | 41.4.0 | 42.0.2.post20191203 | N/A | 46.1.3 |
76+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
77+
| six | >=1.11.0 | >=1.11.0 | 1.12.0 | 1.12.0 | 1.12.0 | 1.12.0 | >=1.12.0 |
78+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
79+
| torch | 0.4.0 | 1.0.0 | 1.1.0 | 1.2.0 | 1.3.1 | 1.4.0 | 1.5.0 |
80+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
81+
| torchvision | 0.2.1 | 0.2.1 | 0.3.0 | 0.4.0a0+9232c4a | 0.4.2 | 0.5.0 | 0.6.0 |
82+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
83+
| Python | 2.7 or 3.5 | 2.7 or 3.6 | 2.7 or 3.6 | 2.7 or 3.6 | 2.7 or 3.6 | 2.7 or 3.6 | 3.6 |
84+
+---------------------------------------+---------------+----------------+---------------+-----------------+---------------------+---------------------+---------------------+
8185

8286
The Docker images extend Ubuntu 16.04.
8387

‎src/sagemaker/pytorch/defaults.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
The default version is no longer updated so as to not break existing workflows.
1919
"""
2020

21-
LATEST_VERSION = "1.4.0"
21+
LATEST_VERSION = "1.5.0"
2222
"""The latest version of PyTorch included in the SageMaker pre-built Docker images."""
2323

2424
PYTHON_VERSION = "py3"

0 commit comments

Comments
 (0)