中国のアリババが展開するクラウドサービス、Alibaba Cloudの記事の第三弾です。
前回の記事はこちら
ついにGPUインスタンスが日本のリージョンで使えるようになりましたね！(https://jp.alibabacloud.com/notice/gpu-notice-20180606)
ということで、今回はGPUインスタンスを使ってディープラーニングのスクリプトを実行してみて、
どれくらいのコストがかかるのかを実験してみたいと思います。

実験

テスト用のスクリプト

ディープラーニング用のフレームワークパッケージであるkerasのリポジトリに置いてあるスクリプトを少し改変したものを使います。
https://github.com/keras-team/keras/blob/master/examples/variational_autoencoder.py
このスクリプトは、Variational Autoencoder という教師なし生成モデルの一つのアルゴリズムを実装したものです。入力データはMNISTという定番の手書き数字画像データです。

生成した画像を実行時に表示するようになっているので、その処理を消してテストに使います(CUIだけで完結させたい)。具体的にはplt.show()を消しました。

また、matplotlibのインポート部分を以下のように変更してます。

import matplotlib
matplotlib.use("Agg")
import matplotlib.pyplot as plt

ローカル環境でのテスト

普段使っているMacBook Airで実行してみます。

スペック

Model Name: MacBook Air
Model Identifier: MacBookAir7,2
Processor Name: Intel Core i7
Processor Speed: 2.2 GHz
Number of Processors: 1
Total Number of Cores: 2
Memory: 8 GB

実行時間

gtime -f %E python3 test_vae.py

13:35.09

約13分35秒でトレーニングが終わりました。

これをGPUインスタンスで実行するとどうなるでしょうか？実験してみましょう！

Alibaba CloudのGPUインスタンスでの実験

1.GPUインスタンスの作成

GPUインスタンスは、以下の設定で作成します。

価格モデル：従量課金
リージョン：アジア東北1(東京)
インスタンスタイプ：ecs.gn5-c4g1.xlarge (4Core、30GB RAM)
イメージ：Ubuntu 16.04 64bit
ストレージ：Ultraクラウドディスク40GB
CPU：Intel Xeon E5-2682 V4(2.5GHz)
GPU：NVIDIA P100
CUDA：バージョン9.1.85/Driver バージョン390.46

CUDAのインストールも「GPUドライバーの自動インストール」にチェックを入れて選択すると、自動でしてくれるみたいです！便利だ。
これはCUDAのインストールスクリプトがインスタンス作成時に実行されるようです。

ちなみにCUDA（クーダ）とはCompute Unified Device Archtectureの略で、NVIDIA社が提供するGPUコンピューティング向けの統合開発環境です。tensorflowなどのディープラーニングフレームワークをGPU環境で使用するときに必要になります。

ここで注意。このあとのスクリプト実行でtensorflowを使用するのですが、これが最新版のCUDA9.1だと動きません。9.0だと現在のtensorflow-gpuの最新版(1.8.0)で動きます。

インスタンスを作成したら、IDをメモしておきます。
今回はi-6we5mrwci0d3tzuqu5q4。

2.cuDNN のインストール

CUDAのディープニューラルネットワーク用ライブラリであるcuDNNをインストールします。

cuDNN v7.1.4 Runtime Library for Ubuntu16.04 (Deb)
cuDNN v7.1.4 Developer Library for Ubuntu16.04 (Deb)
cuDNN v7.1.4 Code Samples and User Guide for Ubuntu16.04 (Deb)

をダウンロード(アカウント登録が必要)し、以下のコマンドでインストールします。

sudo dpkg -i libcudnn7_7.1.4.18-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.1.4.18-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.1.4.18-1+cuda9.0_amd64.deb

3.スクリプトの依存パッケージのインストール

テスト用スクリプトの実行に必要なパッケージを以下のコマンドでインストールします。

apt update
apt upgrade
apt install python3-pip graphviz
pip3 install tensorflow-gpu keras matplotlib pydot

3.テスト用スクリプトの実行

さあ、準備は整いました。実行してみましょう！

time python3 test_vae.py

real	2m20.489s
user	3m6.260s
sys	0m34.004s

約2分20秒で終わりました！めっちゃ速い！
そもそもGPU使ったの初めてなんですが、これはGPU使いたくなりますね。
実験が終わったので、インスタンスをリリースします。

4.値段の確認

しばらく今回の実験が利用状況に反映されるのを待って、
Alibaba Cloudコンソールの右上にある「料金•支払い管理」→料金詳細に移動します。
詳細を見て、今回実験に使ったインスタンスのIDの値段を見てみましょう。
IDはこれでしたね。i-6we5mrwci0d3tzuqu5q4

約186円という結果でした!
一時間未満の利用の場合、料金が１時間単位のため、現在は185.7円が固定でかかるようです。
あとの料金はだいぶ安いですね。
システムディスクも１時間単位なので、それ未満の利用だと40Gだと固定で0.48円かかるようです。
見たところGPUやCPUの使用率は料金に関係ないようなので、フル活用するほどお得ということになるようです。

まとめ

Alibaba CloudのGPUインスタンスをディープラーニングスクリプトの実行に使ってみて、どのくらいの値段がかかるのかを見てみました。想像より値段がかからず驚きました。
普段の研究を考えるとディープラーニングのトレーニングはwebアプリのように日常的に動作し続けるものではないので、一度トレーニングしてインスタンスをリリースできるクラウドは相性がいいと感じました。
オンプレミスでGPU積んだサーバ買うと高いですしね。この実験が誰かの参考になったら幸いです。