Skip to content
Branch: master
Find file Copy path
Find file Copy path
12 contributors

Users who have contributed to this file

@laurenyu @mvsusp @icywang86rui @nadiaya @winstonaws @yangaws @ChoiByungWook @nkconnor @iquintero @jesterhazy @andrewortman @eslesar-aws
74 lines (62 sloc) 6.85 KB

TensorFlow SageMaker Estimators and Models

TensorFlow SageMaker Estimators allow you to run your own TensorFlow training algorithms on SageMaker Learner, and to host your own TensorFlow models on SageMaker Hosting.

Documentation of the previous Legacy Mode versions: 1.4.1, 1.5.0, 1.6.0, 1.7.0, 1.8.0, 1.9.0, 1.10.0

We have added a new format of your TensorFlow training script with TensorFlow version 1.11. This new way gives the user script more flexibility. This new format is called Script Mode, as opposed to Legacy Mode, which is what we support with TensorFlow 1.11 and older versions. In addition we are adding Python 3 support with Script Mode. Last supported version of Legacy Mode will be TensorFlow 1.12. Script Mode is available with TensorFlow version 1.11 and newer. Make sure you refer to the correct version of this README when you prepare your script. You can find the Legacy Mode README here.

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

For information about using TensorFlow with the SageMaker Python SDK, see

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

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.

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 repository, and passing image_name to the TensorFlow Estimator constructor.

For more information on the contents of the images, see the SageMaker TensorFlow containers repository here:

You can’t perform that action at this time.