Amazon SageMaker has Open Sourced TensorFlow 1.6 and Apache MXNet 1.1 Docker Containers with Support for Local Mode, and More Instance Types Across All Modules

Posted on: Apr 4, 2018

You can now use TensorFlow version 1.6 and Apache MXNet version 1.1 with the Amazon SageMaker pre-built deep learning framework containers. Now released as open source on GitHub, these containers can now be downloaded to your local environment for testing before deployment. You can also now use more instance types in Amazon SageMaker, including all Amazon EC2 M5 instance types for training and hosting, 11 more instance types for notebook instances, 3 more instance types for model training, and 16 more instance types for model hosting. 

With the latest versions of the frameworks, you can take advantage of features like AVX instructions for CPUs in TensorFlow, as well as MXNet's Text API, which can be used to build vocabularies and load pre-trained word embeddings. As open source, the TensorFlow and MXNet Docker containers can be customized by adding libraries to the Dockerfile, pre-loading frequently used files and datasets, and changing internal configurations to meet your needs. You can also download the containers to your local environment and use the Amazon SageMaker Python SDK to test your scripts before deploying to Amazon SageMaker training or hosting environments, dramatically accelerating your debugging cycle. And now with support for larger notebook instance types, you can test locally on large datasets or compute intensive jobs. 

These new features in Amazon SageMaker are available today in the US East (N. Virginia), US East (Ohio), EU (Ireland), and U.S. West (Oregon) AWS Regions. Visit the GitHub release notes for more information on TensorFlow 1.6 and Apache MXNet 1.1 features and the Amazon SageMaker Python SDK documentation for more details on how to use these containers locally. Visit the Amazon SageMaker pricing page for details on the added instance types for notebooks, training, and hosting.