Posted On: Feb 27, 2018
You can now use TensorFlow 1.5 and Apache MXNet 1.0 with the pre-built deep learning framework containers that are provided with HAQM SageMaker. HAQM SageMaker also now supports CUDA 9 drivers, for faster performance on HAQM SageMaker ml.p3 instances. Both updates to TensorFlow and MXNet provide improved performance features, among other details that can be found on the respective GitHub repositories for TensorFlow 1.5 and MXNet 1.0.0.
Support for CUDA 9 and the latest TensorFlow and MXNet updates in HAQM SageMaker is available today in the US East (N. Virginia & Ohio), EU (Ireland), and U.S. West (Oregon) AWS regions. To learn more about how to use these newly released containers in HAQM SageMaker, visit the sample notebooks on GitHub.