Skip to content

Commit eefd0c9

Browse files
authored
Support tensorflow 1.9 and bump version to 1.9.2 (#365)
1 parent b73822c commit eefd0c9

File tree

6 files changed

+20
-7
lines changed

6 files changed

+20
-7
lines changed

CHANGELOG.rst

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,11 @@
22
CHANGELOG
33
=========
44

5+
1.9.2
6+
=====
7+
8+
* feature: add support for TensorFlow 1.9
9+
510
1.9.1
611
=====
712

README.rst

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -244,6 +244,8 @@ By using MXNet SageMaker ``Estimators``, you can train and host MXNet models on
244244

245245
Supported versions of MXNet: ``1.2.1``, ``1.1.0``, ``1.0.0``, ``0.12.1``.
246246

247+
We recommend that you use the latest supported version, because that's where we focus most of our development efforts.
248+
247249
For more information, see `MXNet SageMaker Estimators and Models`_.
248250

249251
.. _MXNet SageMaker Estimators and Models: src/sagemaker/mxnet/README.rst
@@ -254,7 +256,9 @@ TensorFlow SageMaker Estimators
254256

255257
By using TensorFlow SageMaker ``Estimators``, you can train and host TensorFlow models on Amazon SageMaker.
256258

257-
Supported versions of TensorFlow: ``1.4.1``, ``1.5.0``, ``1.6.0``, ``1.7.0``, ``1.8.0``.
259+
Supported versions of TensorFlow: ``1.4.1``, ``1.5.0``, ``1.6.0``, ``1.7.0``, ``1.8.0``, ``1.9.0``.
260+
261+
We recommend that you use the latest supported version, because that's where we focus most of our development efforts.
258262

259263
For more information, see `TensorFlow SageMaker Estimators and Models`_.
260264

@@ -268,6 +272,8 @@ By using Chainer SageMaker ``Estimators``, you can train and host Chainer models
268272

269273
Supported versions of Chainer: ``4.0.0``, ``4.1.0``.
270274

275+
We recommend that you use the latest supported version, because that's where we focus most of our development efforts.
276+
271277
For more information about Chainer, see https://github.com/chainer/chainer.
272278

273279
For more information about Chainer SageMaker ``Estimators``, see `Chainer SageMaker Estimators and Models`_.
@@ -280,7 +286,9 @@ PyTorch SageMaker Estimators
280286

281287
With PyTorch SageMaker ``Estimators``, you can train and host PyTorch models on Amazon SageMaker.
282288

283-
Supported versions of PyTorch: ``0.4.0``
289+
Supported versions of PyTorch: ``0.4.0``.
290+
291+
We recommend that you use the latest supported version, because that's where we focus most of our development efforts.
284292

285293
For more information about PyTorch, see https://github.com/pytorch/pytorch.
286294

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def read(fname):
2323

2424

2525
setup(name="sagemaker",
26-
version="1.9.1",
26+
version="1.9.2",
2727
description="Open source library for training and deploying models on Amazon SageMaker.",
2828
packages=find_packages('src'),
2929
package_dir={'': 'src'},

src/sagemaker/tensorflow/README.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ TensorFlow SageMaker Estimators allow you to run your own TensorFlow
66
training algorithms on SageMaker Learner, and to host your own TensorFlow
77
models on SageMaker Hosting.
88

9-
Supported versions of TensorFlow: ``1.4.1``, ``1.5.0``, ``1.6.0``, ``1.7.0``, ``1.8.0``.
9+
Supported versions of TensorFlow: ``1.4.1``, ``1.5.0``, ``1.6.0``, ``1.7.0``, ``1.8.0``, ``1.9.0``.
1010

1111
Training with TensorFlow
1212
~~~~~~~~~~~~~~~~~~~~~~~~
@@ -833,7 +833,7 @@ SageMaker TensorFlow CPU images use TensorFlow built with Intel® MKL-DNN optimi
833833
In certain cases you might be able to get a better performance by disabling this optimization
834834
(`for example when using small models <https://github.com/awslabs/amazon-sagemaker-examples/blob/d88d1c19861fb7733941969f5a68821d9da2982e/sagemaker-python-sdk/tensorflow_iris_dnn_classifier_using_estimators/iris_dnn_classifier.py#L7-L9>`_)
835835

836-
You can disable MKL-DNN optimization for TensorFlow ``1.8.0`` by setting two following environment variables:
836+
You can disable MKL-DNN optimization for TensorFlow ``1.8.0`` and above by setting two following environment variables:
837837

838838
.. code:: python
839839

src/sagemaker/tensorflow/defaults.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,4 +12,4 @@
1212
# language governing permissions and limitations under the License.
1313
from __future__ import absolute_import
1414

15-
TF_VERSION = '1.8'
15+
TF_VERSION = '1.9'

tests/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def sagemaker_local_session(boto_config):
7676

7777

7878
@pytest.fixture(scope='module', params=['1.4', '1.4.1', '1.5', '1.5.0', '1.6', '1.6.0',
79-
'1.7', '1.7.0', '1.8', '1.8.0'])
79+
'1.7', '1.7.0', '1.8', '1.8.0', '1.9', '1.9.0'])
8080
def tf_version(request):
8181
return request.param
8282

0 commit comments

Comments
 (0)