发布于: Mar 26, 2018
适用于 Ubuntu 和 Amazon Linux 的 AWS 深度学习 AMI 现随附为 TensorFlow 1.6 定制的高级优化功能,以便跨 Amazon EC2 实例系列提供高性能训练。
该 AMI 现包含 TensorFlow 1.6 的优化版本,该版本预先配置了 CUDA 9 和 cuDNN 7,以充分利用 P3 GPU 支持的实例上的混合精度训练。这项改进允许 P3 实例为 Machine Learning 应用程序提供更高的训练性能。
此外,TensorFlow 1.6 版本现还包含 Intel 的高级矢量指令 (AVX、AVX-2 和 AVX-512),旨在加速 C5 实例上的矢量和浮点运算。这一 TensorFlow 版本还利用了在深度神经网络训练中用于实现数学例程 CPU 加速的 Intel 深度神经网络数学核心库 (Intel MKL-DNN),现已在 AMI 中得到全面配置。