Skip to content

Commit 3f404df

Browse files
author
Chuyang Deng
committed
update documentation
1 parent 56635c9 commit 3f404df

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

doc/using_tf.rst

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ To train a TensorFlow model by using the SageMaker Python SDK:
4242
Prepare a Script Mode Training Script
4343
======================================
4444

45-
Your TensorFlow training script must be a Python 2.7- or 3.6-compatible source file.
45+
Your TensorFlow training script must be a Python 2.7-, 3.6- or 3.7-compatible source file.
4646

4747
The training script is very similar to a training script you might run outside of SageMaker, but you can access useful properties about the training environment through various environment variables, including the following:
4848

@@ -143,6 +143,11 @@ To use Script Mode, set at least one of these args
143143
- ``py_version='py3'``
144144
- ``script_mode=True``
145145

146+
To use Python 3.7, please specify both of the args:
147+
148+
- ``py_version='py37'``
149+
- ``framework_version='1.15.2'``
150+
146151
When using Script Mode, your training script needs to accept the following args:
147152

148153
- ``model_dir``

src/sagemaker/tensorflow/README.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ The latest containers include the following Python packages:
6969
| tensorflow | 1.15.2 | 2.1.0 |
7070
+--------------------------------+---------------+---------------+
7171

72-
Script Mode TensorFlow Docker images support both Python 2.7 and Python 3.6. The Docker images extend Ubuntu 16.04.
72+
Script Mode TensorFlow Docker images support Python 2.7 and Python 3.6, Python 3.7 for TensorFlow version 1.15.2. The Docker images extend Ubuntu 16.04.
7373

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

0 commit comments

Comments
 (0)