張貼日期: May 15, 2018
Ubuntu 和 Amazon Linux 適用的 AWS 深度學習 AMI 現在隨附 TensorFlow 1.8 的進階最佳化功能,可以更高效的方式訓練 Amazon EC2 C5 和 P3 執行個體。
若為 CPU 型訓練環境,AMI 現在整合了 TensorFlow 1.8 並隨附 Intel 的進階向量指令 (AVX)、SSE 及 FMA 指令集,加速向量和浮點計算。AMI 還完整配備 Intel MKL-DNN,加速 Amazon EC2 C5 執行個體在神經網路訓練中使用的數學常式。對 32 批次大小的 c5.18xlarge 執行個體類型使用最佳化組建時,使用 ImageNet 資料集訓練 ResNet-50 基準測試,速度會是訓練現有 TensorFlow 1.8 二進位資料的 7 倍。
此外,為提升 GPU 型環境的訓練效能,AMI 導入完整配備了 NVIDIA CUDA 9 和 cuDNN 7 的 TensorFlow 1.8 最佳化組建,在支援 Amazon EC2 P3 執行個體的 Volta V100 GPU 上充分發揮混合式精確度訓練的效益。
在您啟用虛擬環境時,深度學習 AMI 會自動部署 TensorFlow 更高效能的組建及其他深度學習框架,例如針對您所選 EC2 執行個體最佳化的 Chainer 和 CNTK。
歡迎善用我們的快速入門教學,開始使用 AWS 深度學習 AMI;如需其他教學和資源,請參閱開發人員指南。您也可以訂閱我們的開發論壇,以掌握最新消息及發文提問。