Skip to content

fix: update TensorFlow script mode dependency list #869

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
Jul 22, 2019
Merged

fix: update TensorFlow script mode dependency list #869

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
765480f
fix: update TensorFlow script mode dependency list
icywang86rui Jun 21, 2019
3f9bba8
Merge branch 'master' into upldate-doc
icywang86rui Jul 19, 2019
0e464e5
Merge branch 'master' into upldate-doc
icywang86rui Jul 22, 2019
0838051
update tfs ei version in README.rst
icywang86rui Jul 22, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion README.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,7 @@ By using TensorFlow SageMaker Estimators, you can train and host TensorFlow mode

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``.

Supported versions of TensorFlow for Elastic Inference: ``1.11.0``, ``1.12.0``.
Supported versions of TensorFlow for Elastic Inference: ``1.11.0``, ``1.12.0``, ``1.13.0``

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

Expand Down
75 changes: 39 additions & 36 deletions src/sagemaker/tensorflow/README.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ Documentation of the previous Legacy Mode versions: `1.4.1 <https://github.com/a
| 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>`_. |
+-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------+

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

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

Expand All @@ -29,42 +29,45 @@ SageMaker TensorFlow Docker containers

The containers include the following Python packages:

+--------------------------------+---------------+-------------------+
| Dependencies | Script Mode | Legacy Mode |
+--------------------------------+---------------+-------------------+
| boto3 | Latest | Latest |
+--------------------------------+---------------+-------------------+
| botocore | Latest | Latest |
+--------------------------------+---------------+-------------------+
| CUDA (GPU image only) | 9.0 | 9.0 |
+--------------------------------+---------------+-------------------+
| numpy | Latest | Latest |
+--------------------------------+---------------+-------------------+
| Pillow | Latest | Latest |
+--------------------------------+---------------+-------------------+
| scipy | Latest | Latest |
+--------------------------------+---------------+-------------------+
| sklean | Latest | Latest |
+--------------------------------+---------------+-------------------+
| h5py | Latest | Latest |
+--------------------------------+---------------+-------------------+
| pip | 18.1 | 18.1 |
+--------------------------------+---------------+-------------------+
| curl | Latest | Latest |
+--------------------------------+---------------+-------------------+
| tensorflow | 1.12.0 | 1.12.0 |
+--------------------------------+---------------+-------------------+
| tensorflow-serving-api | 1.12.0 | None |
+--------------------------------+---------------+-------------------+
| sagemaker-containers | >=2.3.5 | >=2.3.5 |
+--------------------------------+---------------+-------------------+
| sagemaker-tensorflow-container | 1.0 | 1.0 |
+--------------------------------+---------------+-------------------+
| Python | 2.7 or 3.6 | 2.7 |
+--------------------------------+---------------+-------------------+
+--------------------------------+---------------+
| Dependencies | Script Mode |
+--------------------------------+---------------+
| awscli | 1.16.130 |
+--------------------------------+---------------+
| boto3 | Latest |
+--------------------------------+---------------+
| botocore | Latest |
+--------------------------------+---------------+
| h5py | 2.9.0 |
+--------------------------------+---------------+
| keras | 2.2.4 |
+--------------------------------+---------------+
| mpi4py | 3.0.1 |
+--------------------------------+---------------+
| numpy | 1.16.2 |
+--------------------------------+---------------+
| pandas | 0.24.2 |
+--------------------------------+---------------+
| pip | 19.1.1 |
+--------------------------------+---------------+
| Pillow | 5.4.1 |
+--------------------------------+---------------+
| Python | 2.7 or 3.6 |
+--------------------------------+---------------+
| requests | 2.21.0 |
+--------------------------------+---------------+
| sagemaker-containers | >=2.4.6 |
+--------------------------------+---------------+
| sagemaker-tensorflow-container | Latest |
+--------------------------------+---------------+
| scipy | 1.2.1 |
+--------------------------------+---------------+
| sklean | 0.20.3 |
+--------------------------------+---------------+
| tensorflow | 1.13.0 |
+--------------------------------+---------------+

Legacy Mode TensorFlow Docker images support Python 2.7. Script Mode TensorFlow Docker images support both Python 2.7
and Python 3.6. The Docker images extend Ubuntu 16.04.
Script Mode TensorFlow Docker images support both Python 2.7 and Python 3.6. The Docker images extend Ubuntu 16.04.

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.
Alternatively, you can build your own image by following the instructions in the SageMaker TensorFlow containers
Expand Down