Download
This package contains SageMaker-specific extensions to TensorFlow, including the PipeModeDataset class, that allows SageMaker Pipe Mode channels to be read using TensorFlow Datasets. This package supports Python 3.7-3.9 and TensorFlow versions 1.7 and higher, including 2.0-2.9.1. For TensorFlow 1.x support, see the master branch. sagemaker-tensorflow releases for all supported versions are available on PyPI. SageMaker TensorFlow extensions builds on Python 3.4-3.9 in Linux with a TensorFlow version >= 1.7. Older versions of TensorFlow are not supported. Please make sure to checkout the branch of sagemaker-tensorflow-extensions that matches your TensorFlow version. 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.
Features
- You can build SageMaker TensorFlow into a docker image
- You can also install sagemaker-tensorflow for a specific version of TensorFlow
- Build and install from source
- The SageMaker TensorFlow build depends on cmake
- Build for a specific TensorFlow version
- SageMaker TensorFlow extensions builds on Python 3.4-3.9 in Linux
Project Samples
