張貼日期: Apr 10, 2018
Ubuntu 和 Amazon Linux 適用的 AWS 深度學習 AMI 現在隨附專為 TensorFlow 1.7 定製的進階最佳化，可提供跨 Amazon EC2 C5 和 P3 執行個體的高效能訓練。
對於 CPU 型訓練環境，AMI 現在引入 TensorFlow 1.7 並隨附 Intel 的進階向量指令 (AVX)、SSE 及 FMA 指令集，以加速向量和浮點計算。AMI 還完整配備 Intel MKL-DNN，可透過為 Amazon EC2 C5 執行個體提供支援的 Intel Xeon Platinum 處理器加速神經網路訓練中使用的數學常式。透過 c5.18xlarge 執行個體類型的最佳化建置，使用合成的 ImageNet 資料集訓練 ResNet-50 基準的速度比使用庫存 TensorFlow 1.7 二進位檔進行訓練的速度更快 9.8 倍。
此外，為提升 GPU 型環境的訓練效能，AMI 引入完整配備 NVIDIA CUDA 9 和 cuDNN 7 的 TensorFlow 1.7 的最佳化建置，可透過為 Amazon EC2 P3 執行個體提供支援的 Volta V100 GPU 充分利用混合的精確度訓練。
最後，此版本還引入 TensorBoard 1.7 以協助您視覺化呈現模型訓練及偵錯，並引入 TensorFlow Serving 1.6 以快速為您的訓練模型提供推斷端點的原型。AMI 還引入 Microsoft Cognitive Toolkit 2.5，可提升效能和修正錯誤。