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