張貼日期: Oct 18, 2017

今天,我們很高興宣布 AWS 深度學習 AMI 的更新。

AWS 深度學習 AMI 可讓您按一下按鈕即可在 AWS 上運作完整的深度學習環境,現可支援 PyTorch、Keras 1.2 和 2.0,同時搭配諸如 TensorFlow、Caffe2 及 Apache MXNet 等熱門機器學習架構。

使用 PyTorch 快速建立原型

AMI 現納入 PyTorch 0.2.0,讓開發人員能夠用 Python 建立動態類神經網路,非常適合文字和時間序列等動態輸入。開發人員可使用這些新手和進階教學課程,包括設立 PyTorch 的分散式訓練,快速開始著手。

改善 Keras 支援

AMI 現可支援最新版本的 Keras v2.0.8,您的 Keras 代碼預設 TensorFlow 做為後端來執行;您也可換至其他受支援的後端,例如 Theano 和 CNTK。我們也納入 Keras 1.2.2 的修改版本,可執行於 Apache MXNet 後端,提升訓練效能。

使用最新架構進行預先安裝及設定

此 AMI 版本支援下列架構的最新版本:

  • 搭載 Gluon 的 Apache MXNet 0.11.0
  • TensorFlow 1.3.0
  • Caffe2 0.8.0
  • Caffe1.0
  • PyTorch 0.2.0
  • 預設 TensorFlow 為後端的 Keras 2.0.8
  • 預設 MXNet 為後端的 Keras 1.2.2 (DMLC 延伸)
  • Theano 0.9.0
  • CNTK 2.0
  • Torch (主要分支)

套裝也包含下列提升 GPU 速度的預先設定程式庫:

  • CUDA 工具組 8.0
  • cuDNN 5.1
  • NVidia 驅動程式 375.66
  • NCCL 2.0

試用 Gluon

最後,AMI 納入全新開放原始碼深度學習介面 Gluon,讓開發人員輕鬆快速建置機器學習模型,不用犧牲效能。您可以在我們的推出公告進一步了解 Gluon 的詳細資訊,並利用帶有範本程式碼的 50 本筆記本立刻開始著手。

您可以在 AWS Marketplace 按一下,即可啟動 UbuntuAmazon Linux 的 AWS 深度學習 AMI,或者遵循逐步操作指南開始著手進行並啟動第一本筆記本。

建模愉快!

附註:Keras 支援筆記。

您可以使用 Conda 虛擬環境,在 Keras 1 和 Keras 2 間進行切換。預設執行 Keras 2;如要切換至 Keras 1 及 MXNet 後端,請使用下列命令:

針對 Python 2 使用者:

source ~/src/anaconda3/bin/activate keras1.2_p2

針對 Python 3 使用者:

 source ~/src/anaconda3/bin/activate keras1.2_p3

接著,您可以從此虛擬環境內部匯入並執行 Keras 1.2.2,正常命令如下:

import keras

您可以前往 Conda 入門指南,進一步了解 Conda 及其用於管理虛擬環境的命令列介面。