From 765480febced516c8e56ee7232efcf605b6f5a9c Mon Sep 17 00:00:00 2001 From: Rui Wang Date: Fri, 21 Jun 2019 11:17:43 -0700 Subject: [PATCH 1/2] fix: update TensorFlow script mode dependency list --- src/sagemaker/tensorflow/README.rst | 75 +++++++++++++++-------------- 1 file changed, 39 insertions(+), 36 deletions(-) diff --git a/src/sagemaker/tensorflow/README.rst b/src/sagemaker/tensorflow/README.rst index cae54f7185..3533288eed 100644 --- a/src/sagemaker/tensorflow/README.rst +++ b/src/sagemaker/tensorflow/README.rst @@ -20,7 +20,7 @@ Documentation of the previous Legacy Mode versions: `1.4.1 `_. | +-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------+ -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. @@ -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 From 0838051de261555216adfdeb010eaa12c8ad1b5a Mon Sep 17 00:00:00 2001 From: Rui Wang Date: Mon, 22 Jul 2019 09:34:12 -0700 Subject: [PATCH 2/2] update tfs ei version in README.rst --- README.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.rst b/README.rst index eff60c8941..e1e0a02864 100644 --- a/README.rst +++ b/README.rst @@ -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.