Edited at

NVIDIA GPU-REST-Engine を AWS で動かしてみよう

More than 3 years have passed since last update.

Chromeで見ると画像の色合いが変なのでSafariやFirefoxで見てください。


はじめに


GPU REST Engine

GPU REST Engine は、Deep Learning Frameworkなど CUDAアプリケーションを "REST API" で操作する仕組みであり、NVIDIAが提供しています。

とは言え、AWS API GATEWAY などと違って、結構自分でコード書く必要はあります。


GPU REST Engine on GitHub

GitHub上のGPU REST Engine は、 Deep Learning のフレームワークの一つである CAFFE の学習済みのモデルを用い、画像に写っている動物の種類を判別するサンプルです。

例えば下記のようにうさぎの写真を入れると、Angora rabbit の可能性が 0.9998 と Json で返ってきます。

Kobito.I4MraJ.png

GPU REST Engine は、主に Docker, go-langで構成されており、 Web Server は、main.goというコードの中で起動されています。


このQiitaのページにかかれていること

このQiitaのドキュメントは、AWS の GPU インスタンスを用いて、GitHub上にある GPU REST Engineのサンプルを動作させます。


Hands-on 内容


参考にしたドキュメント


このQiitaの情報は、おいらがプライベートで作っているもので、NVIDIAとは無関係です。個人責任でお試しください。インスタンス等の消し忘れ等ご注意ください。