Skip to content

Commit 7b9ad5c

Browse files
icywang86ruilaurenyu
icywang86rui
authored and
laurenyu
committed
fix: update TensorFlow script mode dependency list (#869)
1 parent af4b3e3 commit 7b9ad5c

File tree

2 files changed

+40
-37
lines changed

2 files changed

+40
-37
lines changed

‎README.rst

+1-1
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,7 @@ By using TensorFlow SageMaker Estimators, you can train and host TensorFlow mode
191191

192192
Supported versions of TensorFlow: ``1.4.1``, ``1.5.0``, ``1.6.0``, ``1.7.0``, ``1.8.0``, ``1.9.0``, ``1.10.0``, ``1.11.0``, ``1.12.0``, ``1.13.1``.
193193

194-
Supported versions of TensorFlow for Elastic Inference: ``1.11.0``, ``1.12.0``.
194+
Supported versions of TensorFlow for Elastic Inference: ``1.11.0``, ``1.12.0``, ``1.13.0``
195195

196196
We recommend that you use the latest supported version, because that's where we focus most of our development efforts.
197197

‎src/sagemaker/tensorflow/README.rst

+39-36
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ Documentation of the previous Legacy Mode versions: `1.4.1 <https://github.com/a
2020
| You can find the Legacy Mode README `here <https://github.com/aws/sagemaker-python-sdk/tree/v1.12.0/src/sagemaker/tensorflow#tensorflow-sagemaker-estimators-and-models>`_. |
2121
+-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------+
2222

23-
Supported versions of TensorFlow for Elastic Inference: ``1.11.0``, ``1.12.0``.
23+
Supported versions of TensorFlow for Elastic Inference: ``1.11``, ``1.12``, ``1.13``.
2424

2525
For information about using TensorFlow with the SageMaker Python SDK, see https://sagemaker.readthedocs.io/en/stable/using_tf.html.
2626

@@ -29,42 +29,45 @@ SageMaker TensorFlow Docker containers
2929

3030
The containers include the following Python packages:
3131

32-
+--------------------------------+---------------+-------------------+
33-
| Dependencies | Script Mode | Legacy Mode |
34-
+--------------------------------+---------------+-------------------+
35-
| boto3 | Latest | Latest |
36-
+--------------------------------+---------------+-------------------+
37-
| botocore | Latest | Latest |
38-
+--------------------------------+---------------+-------------------+
39-
| CUDA (GPU image only) | 9.0 | 9.0 |
40-
+--------------------------------+---------------+-------------------+
41-
| numpy | Latest | Latest |
42-
+--------------------------------+---------------+-------------------+
43-
| Pillow | Latest | Latest |
44-
+--------------------------------+---------------+-------------------+
45-
| scipy | Latest | Latest |
46-
+--------------------------------+---------------+-------------------+
47-
| sklean | Latest | Latest |
48-
+--------------------------------+---------------+-------------------+
49-
| h5py | Latest | Latest |
50-
+--------------------------------+---------------+-------------------+
51-
| pip | 18.1 | 18.1 |
52-
+--------------------------------+---------------+-------------------+
53-
| curl | Latest | Latest |
54-
+--------------------------------+---------------+-------------------+
55-
| tensorflow | 1.12.0 | 1.12.0 |
56-
+--------------------------------+---------------+-------------------+
57-
| tensorflow-serving-api | 1.12.0 | None |
58-
+--------------------------------+---------------+-------------------+
59-
| sagemaker-containers | >=2.3.5 | >=2.3.5 |
60-
+--------------------------------+---------------+-------------------+
61-
| sagemaker-tensorflow-container | 1.0 | 1.0 |
62-
+--------------------------------+---------------+-------------------+
63-
| Python | 2.7 or 3.6 | 2.7 |
64-
+--------------------------------+---------------+-------------------+
32+
+--------------------------------+---------------+
33+
| Dependencies | Script Mode |
34+
+--------------------------------+---------------+
35+
| awscli | 1.16.130 |
36+
+--------------------------------+---------------+
37+
| boto3 | Latest |
38+
+--------------------------------+---------------+
39+
| botocore | Latest |
40+
+--------------------------------+---------------+
41+
| h5py | 2.9.0 |
42+
+--------------------------------+---------------+
43+
| keras | 2.2.4 |
44+
+--------------------------------+---------------+
45+
| mpi4py | 3.0.1 |
46+
+--------------------------------+---------------+
47+
| numpy | 1.16.2 |
48+
+--------------------------------+---------------+
49+
| pandas | 0.24.2 |
50+
+--------------------------------+---------------+
51+
| pip | 19.1.1 |
52+
+--------------------------------+---------------+
53+
| Pillow | 5.4.1 |
54+
+--------------------------------+---------------+
55+
| Python | 2.7 or 3.6 |
56+
+--------------------------------+---------------+
57+
| requests | 2.21.0 |
58+
+--------------------------------+---------------+
59+
| sagemaker-containers | >=2.4.6 |
60+
+--------------------------------+---------------+
61+
| sagemaker-tensorflow-container | Latest |
62+
+--------------------------------+---------------+
63+
| scipy | 1.2.1 |
64+
+--------------------------------+---------------+
65+
| sklean | 0.20.3 |
66+
+--------------------------------+---------------+
67+
| tensorflow | 1.13.0 |
68+
+--------------------------------+---------------+
6569

66-
Legacy Mode TensorFlow Docker images support Python 2.7. Script Mode TensorFlow Docker images support both Python 2.7
67-
and Python 3.6. The Docker images extend Ubuntu 16.04.
70+
Script Mode TensorFlow Docker images support both Python 2.7 and Python 3.6. The Docker images extend Ubuntu 16.04.
6871

6972
You can select version of TensorFlow by passing a ``framework_version`` keyword arg to the TensorFlow Estimator constructor. Currently supported versions are listed in the table above. You can also set ``framework_version`` to only specify major and minor version, e.g ``'1.6'``, which will cause your training script to be run on the latest supported patch version of that minor version, which in this example would be 1.6.0.
7073
Alternatively, you can build your own image by following the instructions in the SageMaker TensorFlow containers

0 commit comments

Comments
 (0)