发布于: Apr 30, 2018

AWS Deep Learning AMI 现在提供对 TensorFlow 1.7 的高级优化方案,以交付更高性能的 Amazon EC2 实例训练。

对于基于 CPU 的训练场景,AMI 现在提供使用 Intel 先进矢量指令 (AVX)、SSE 和 FMA 指令集构建的 TensorFlow 1.7,以加快矢量和浮点计算。AMI 还全面配置了 Intel MKL-DNN,来加速 Amazon EC2 C3 和 C4 实例所用的 Intel Xeon 处理器的中性网络训练中使用的数学例程。 

此外,为了提高基于 GPU 的场景的训练性能,AMI 还提供一个经过优化的 TensorFlow 1.7(全面配置了 NVIDIA CUDA 9 和 cuDNN 7),可对 Amazon EC2 P3 实例所用的 Volta V100 GPU 实现混合精度训练。

当您首次激活 TensorFlow 虚拟环境时,该 AMI 会自动部署针对您选择的 EC2 实例进行优化了的 TensorFlow。

您可在 EC2 实例启动向导中的“步骤 1:选择 Amazon 系统映像 (AMI)”的快速入门部分找到您选择的 Deep Learning AMI。另请参阅我们的文档指南,以帮助您为自己的项目选择正确的 AMI、了解简单教程并获得其他资源。