SageMaker specific extensions to TensorFlow, for Python 2.7, 3.4-3.6 and TensorFlow versions 1.7, 1.8, and 1.9. This package includes the PipeModeDataset class, that allows SageMaker Pipe Mode channels to be read using TensorFlow DataSets.
Install
You can build SageMaker TensorFlow into a docker image with the following command:
pip install sagemaker-tensorflow
You can also install sagemaker-tensorflow for a specific version of TensorFlow. The following command will install sagemaker-tensorflow for TensorFlow 1.7:
pip install "sagemaker-tensorflow>=1.7,<1.8"
Build and install from source
The SageMaker TensorFlow build depends on the following:
cmake
tensorflow
curl-dev
To install these run:
pip install cmake tensorflow
On Amazon Linux, curl-dev can be installed with:
yum install curl-dev
On Ubuntu, curl-dev can be installed with:
apt-get install libcurl4-openssl-dev
To build and install this package, run:
pip install .
in this directory.
To build in a SageMaker docker image, you can use the following RUN command in your Dockerfile:
RUN git clone https://github.com/aws/sagemaker-tensorflow-extensions.git && \
cd sagemaker-tensorflow-extensions && \
pip install . && \
cd .. && \
rm -rf sagemaker-tensorflow-extensions
Building for a specific TensorFlow version
Release branching is used to track different versions of TensorFlow. Tensorflow versions 1.7 and 1.8 are supported. To build for a specific release of TensorFlow, checkout the release branch prior to running a pip install. For example, to build for TensorFlow 1.7, you can run the following command in your Dockerfile:
RUN git clone https://github.com/aws/sagemaker-tensorflow-extensions.git && \
cd sagemaker-tensorflow-extensions && \
git checkout 1.7 && \
pip install . && \
cd .. && \
rm -rf sagemaker-tensorflow-extensions
Requirements
SageMaker TensorFlow extensions builds on Python 2.7 in Linux, with either TensorFlow 1.7, 1.8, and 1.9. Please make sure to checkout the branch of sagemaker-tensorflow-extensions that matches your TensorFlow version installed.
SageMaker Pipe Mode
SageMaker Pipe Mode is a mechanism for providing S3 data to a training job via Linux fifos. Training programs can read from the fifo and get high-throughput data transfer from S3, without managing the S3 access in the program itself.
SageMaker Pipe Mode is enabled when a SageMaker Training Job is created. Multiple S3 datasets can be mapped to individual fifos, configured in the training request. Pipe Mode is covered in more detail in the SageMaker documentation: https://docs.aws.amazon.com/sagemaker/latest/dg/your-algorithms-training-algo.html#your-algorithms-training-algo-running-container-inputdataconfig
Using the PipeModeDataset
The PipeModeDataset is a TensorFlow Dataset for reading SageMaker Pipe Mode channels. After installing the sagemaker tensorflow extensions package, the PipeModeDataset can be imported from a moduled named sagemaker_tensorflow.
To construct a PipeModeDataset that reads TFRecord encoded records from a “training” channel, do the following:
from sagemaker_tensorflow import PipeModeDataset
ds = PipeModeDataset(channel='training', record_format='TFRecord')A PipeModeDataset should be created for a SageMaker PipeMode channel. Each channel corresponds to a single S3 dataset, configured when the training job is created. You can create multiple PipeModeDataset instances over different channels to read from multiple S3 datasets in the same training program.
A PipeModeDataset can read TFRecord, RecordIO, or text line records, by using the record_format constructor argument. The record_format kwarg can be set to either RecordIO, TFRecord, or TextLine to differentiate between the three encodings. RecordIO is the default.
A PipeModeDataset is a regular TensorFlow Dataset and as such can be used in TensorFlow input processing pipelines, and in TensorFlow Estimator input_fn definitions. All Dataset operations are supported on PipeModeDataset. The following code snippet shows how to create a batching and parsing Dataset that reads data from a SageMaker Pipe Mode channel:
features = {
'data': tf.FixedLenFeature([], tf.string),
'labels': tf.FixedLenFeature([], tf.int64),
}
def parse(record):
parsed = tf.parse_single_example(record, features)
return ({
'data': tf.decode_raw(parsed['data'], tf.float64)
}, parsed['labels'])
ds = PipeModeDataset(channel='training', record_format='TFRecord')
num_epochs = 20
ds = ds.repeat(num_epochs)
ds = ds.prefetch(10)
ds = ds.map(parse, num_parallel_calls=10)
ds = ds.batch(64)License
SageMaker TensorFlow is licensed under the Apache 2.0 License. It is copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved. The license is available at: http://aws.amazon.com/apache2.0/
1 maintainer
Release files for sagemaker-tensorflow 1.10.0.1.0.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distributions (wheels)
| File | Reset | |||
|---|---|---|---|---|
| sagemaker_tensorflow-1.10.0.1.0.0-cp36-cp36m-manylinux1_x86_64.whl | CPython 3.6 | CPython 3.6 pymalloc | Linux glibc 2.5+ x86-64 | Details |
| sagemaker_tensorflow-1.10.0.1.0.0-cp35-cp35m-manylinux1_x86_64.whl | CPython 3.5 | CPython 3.5 pymalloc | Linux glibc 2.5+ x86-64 | Details |
| sagemaker_tensorflow-1.10.0.1.0.0-cp34-cp34m-manylinux1_x86_64.whl | CPython 3.4 | CPython 3.4 pymalloc | Linux glibc 2.5+ x86-64 | Details |
| sagemaker_tensorflow-1.10.0.1.0.0-cp27-cp27mu-manylinux1_x86_64.whl | CPython 2.7 | CPython 2.7 pymalloc wide-unicode | Linux glibc 2.5+ x86-64 | Details |
Total release size: 2.3 MB
Release files / sagemaker_tensorflow-1.10.0.1.0.0-cp36-cp36m-manylinux1_x86_64.whl
| Download URL | sagemaker_tensorflow-1.10.0.1.0.0-cp36-cp36m-manylinux1_x86_64.whl |
|---|---|
| Size | 574.4 kB |
| Tags | CPython 3.6 CPython 3.6 pymalloc Linux glibc 2.5+ x86-64 |
|
SHA-256 checksum How to use checksums |
3f877ab09b84f1bff6db1aa07f7be0ececc2465f0f5efb4e89ef54da89ccde7a
|
|
BLAKE2b-256 checksum How to use checksums |
e3cb2fcbe50eae44ece00df78640b0d6e39938b25d0dd516ab900b3e8edb11d8
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/1.11.0 pkginfo/1.4.2 requests/2.19.1 setuptools/28.8.0 requests-toolbelt/0.8.0 tqdm/4.24.0 CPython/3.6.2
|
Release files / sagemaker_tensorflow-1.10.0.1.0.0-cp35-cp35m-manylinux1_x86_64.whl
| Download URL | sagemaker_tensorflow-1.10.0.1.0.0-cp35-cp35m-manylinux1_x86_64.whl |
|---|---|
| Size | 574.4 kB |
| Tags | CPython 3.5 CPython 3.5 pymalloc Linux glibc 2.5+ x86-64 |
|
SHA-256 checksum How to use checksums |
a730bae7e5ca0eec178f42dbae22b9e281e04beb204ce6dad529d1024be808a9
|
|
BLAKE2b-256 checksum How to use checksums |
bd4378e350f872359fc49e82649bd90f65f1a5fe4bfc4caf4814b3b50b76ef32
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/1.11.0 pkginfo/1.4.2 requests/2.19.1 setuptools/20.10.1 requests-toolbelt/0.8.0 tqdm/4.24.0 CPython/3.5.2
|
Release files / sagemaker_tensorflow-1.10.0.1.0.0-cp34-cp34m-manylinux1_x86_64.whl
| Download URL | sagemaker_tensorflow-1.10.0.1.0.0-cp34-cp34m-manylinux1_x86_64.whl |
|---|---|
| Size | 574.4 kB |
| Tags | CPython 3.4 CPython 3.4 pymalloc Linux glibc 2.5+ x86-64 |
|
SHA-256 checksum How to use checksums |
0ef1ed9172cee7e002e32a0887faa5fc75bd2e3f4e756348f11876dd919a5214
|
|
BLAKE2b-256 checksum How to use checksums |
689b4fffc51925711f511b942cf54003c42f1a47e347d2a1692fa164b1e7c8cc
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/1.11.0 pkginfo/1.4.2 requests/2.19.1 setuptools/39.1.0 requests-toolbelt/0.8.0 tqdm/4.24.0 CPython/3.4.7
|
Release files / sagemaker_tensorflow-1.10.0.1.0.0-cp27-cp27mu-manylinux1_x86_64.whl
| Download URL | sagemaker_tensorflow-1.10.0.1.0.0-cp27-cp27mu-manylinux1_x86_64.whl |
|---|---|
| Size | 577.2 kB |
| Tags | CPython 2.7 CPython 2.7 pymalloc wide-unicode Linux glibc 2.5+ x86-64 |
|
SHA-256 checksum How to use checksums |
d425a1e993cd3a8dbf3f7b7f810b59ce94fabfb31c7373be03bab3de637434ef
|
|
BLAKE2b-256 checksum How to use checksums |
5c61cec06a464ab100b27f4ebd880f3086fed4960bf82ba407c0b353a2226785
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/1.11.0 pkginfo/1.4.2 requests/2.18.4 setuptools/39.1.0 requests-toolbelt/0.8.0 tqdm/4.24.0 CPython/2.7.10
|
