Tesla P100 PCIe 16GB のCaffeでの学習時間(AlexNet)

学習用データは前回の記事「CAFFEの学習時間をGTX 1080とNVIDIA TITAN Xで比較する」と同じものを使いました。具体的には「画像分類のコンテスト等て使われているカラーで265x256サイズの学習用画像を手軽に用意したかったので、CIFAR10(カラー, 32x32)を、Digitsで256x256に解像度を上げて使いました。1epoch37,500枚の画像を30epoch(1,125,000枚)学習する時間を測定しました。」です。

なんと19分53秒でNVIDIA TITAN Xの37分56秒に比べて圧倒的に高速です。Float16はまだ使われてないと思うのですが。2GPU では13分10秒でした。2GPUで1.5倍なら十分2GPUにする価値はありますね。 

 P100 * 1                                     P100 * 2

AlexNetP100
AlexNet2xP100