2019年12月5日にアップデートされたDeep Learning AMI(DLAMI)についての解説です。
ちなみに、AWSのサイトでは AWS 深層学習 AMI と Deep Learning AMI という2つの表現が出てきますが、どちらも同じものを指しています。
公式の出している開発者ガイドに詳しい使い方などがわかりやすく載っているので、この記事では軽い概要の紹介をしていきます。
Deep Learning AMIとは
まず、Deep Learning AMI(DLAMI)には
- Deep Learning AMIとConda
- Deep Learning Base AMI
の2つがあります。
どちらもDeep Learning向けのAMIですが、違いは機械学習Frameworkがインストールされているか否かです。
Deep Learning AMIとConda の方ではAnaconda仮想環境でFrameworkを個別に管理しており、そのまま使用する事ができるようです。
Deep Learning Base AMI の方はFrameworkは入っておらず、GPUを使ってFrameworkを使用するのに必要なcuDNNなどの設定がされており、必要なFrameworkのインストールをするだけになっているようです。
詳しくは次のようになっています。(2020年1月6日時点 クイックスタートでの検索で出てくるAMI)
更に詳しい情報はマーケットプレイスでAMI IDを検索すれば調べる事ができます。
| AMI | AMI ID | MXNet | Tensorflow | PyTorch-1.3.1 | Keras-2.2 | Caffe | Chainer | CNTK | Theano | NVIDIA CUDA | cuDNN | NCCL | Intel MKL-DNN | Docker & NVIDIA-Docker | GPU Driver |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Deep Learning AMI (Ubuntu 18.04) Version 26.0 | ami-07729b5941107618c | ○(1.6.0rc0) | ○(2.0 &1.15) | ○ | ○(Neuron) | – | ○ | – | – | ○ | ○ | ○ | ○ | ○ | ○ |
| Deep Learning AMI (Ubuntu 16.04) Version 26.0 | ami-039d0c16bef5e0155 | ○(1.6.0rc0) | ○(2.0 &1.15) | ○ | ○(Neuron) | ○ | ○ | ○ | ○ | ○ | ○ | ○ | ○ | ○ | ○ |
| Deep Learning AMI (Amazon Linux) Version 26.0 | ami-07663a5b4e31e9963 | ○(1.6.0rc0) | ○(2.0 &1.15) | ○ | ○(Elastic Inference) | ○ | ○ | ○ | ○ | ○ | ○ | ○ | ○ | ○ | ○ |
| Deep Learning AMI (Amazon Linux 2) Version 26.0 | ami-0f7912d4a98edaa3d | ○(1.6.0rc0) | ○(2.0 &1.15) | ○ | ○(Neuron) | – | ○ | – | – | ○ | ○ | ○ | ○ | ○ | ○ |
| Deep Learning AMI (Microsoft Windows Server 2016) | ami-017b0f5a3924d707b | ○(0.12 RC) | ○(1.4) | – | – | ○ | – | – | – | ○ | ○ | – | – | – | ○ |
| Deep Learning Base AMI (Ubuntu 18.04) Version 21.0 | ami-096da10df231117a7 | – | – | – | – | – | – | – | – | ○ | ○ | ○ | ○ | – | ○ |
| Deep Learning Base AMI (Ubuntu 16.04) Version 21.0 | ami-04ead0142968f9046 | – | – | – | – | – | – | – | – | ○ | ○ | ○ | ○ | – | ○ |
| Deep Learning Base AMI (Amazon Linux) Version 21.0 | ami-0e472660521e8b8b7 | – | – | – | – | – | – | – | – | ○ | ○ | ○ | ○ | – | ○ |
| Deep Learning Base AMI (Amazon Linux 2) Version 21.0 | ami-05af204c112746305 | – | – | – | – | – | – | – | – | ○ | ○ | ○ | ○ | – | ○ |
料金
AMIそのものに料金はかかりません。
推奨環境
GPUインスタンスの場合
| インスタンスタイプ | GPUの種類 | 最大GPU数 |
|---|---|---|
| P3 インスタンス | NVIDIA Tesla V100 GPU | 8 |
| P2 インスタンス | NVIDIA K80 GPU | 16 |
| G3 インスタンス | NVIDIA Tesla M60 GPU | 4 |
CPUインスタンスの場合
| インスタンスタイプ | 最大CPU数 |
|---|---|
| C5 インスタンス | 72 |
| C4 インスタンス | 36 |
※ Caffeを使う場合はGPUインスタンスを使ってください。DLAMIのCaffeはGPUインスタンスのみをサポートしているそうです。
実際に使ってみました
- Deep Learning AMIを使ってみた(失敗)
- Deep Learning AMIをCPUインスタンスで使ってみた
- Deep Learning AMIをGPUインスタンスで使ってみた
- Deep Learning Base AMIをGPUインスタンスで使ってみた => 後日公開予定
