sony製の深層学習ライブラリ。
モォ動かしてる人がいるじゃないか。
http://qiita.com/miumiu0917/items/c8895c488aed8834ff1b
ってことで一応動かした。
動的計算グラフって書いてるし、chainer like。
sudo pip install nnabla
sudo pip install nnabla-ext-cuda
git clone https://github.com/sony/nnabla.git
mnistサンプル
cd examples/vision/mnist/
python classification.py
2017-06-27 22:52:33,967 [nnabla][INFO]: Parameter save (hdf5): tmp.monitor/params_000000.h5
2017-06-27 22:52:34,625 [nnabla][INFO]: iter=9 {Training loss}=2.29262495041
2017-06-27 22:52:34,625 [nnabla][INFO]: iter=9 {Training error}=0.8546875
2017-06-27 22:52:35,254 [nnabla][INFO]: iter=19 {Training loss}=1.95494496822
2017-06-27 22:52:35,254 [nnabla][INFO]: iter=19 {Training error}=0.6015625
...
imagenetサンプル
tiny-imagenet-200.zipってのダウンロードするけど、cpuで学習には結構時間かかる。
cd examples/vision/imagenet/
python classification.py -a4 -b64 -L34
2017-06-27 23:25:23,584 [nnabla][INFO]: Initializing CPU extension...
2017-06-27 23:25:24,034 [nnabla][INFO]: Running in cpu
2017-06-27 23:25:24,035 [nnabla][INFO]: DataSource with shuffle(True)
2017-06-27 23:25:24,334 [nnabla][INFO]: Using DataIterator
2017-06-27 23:25:24,336 [nnabla][INFO]: DataSource with shuffle(True)
2017-06-27 23:25:24,445 [nnabla][INFO]: Using DataIterator
2017-06-27 23:25:25,064 [nnabla][INFO]: Parameter save (hdf5): tmp.monitor.imagenet/param_000000.h5
感想
cpuで動いたけどgpuで動かない。cudnn6.0の最新にしろってエラーのようだ。
めんどくさいのでまた今度にする。
ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory
cpuではすぐに動いた。cpuだとc++で高速に動いてそう。
cudaとの繋ぎこみはchainerのcupy見たく別ライブラリになってる。
https://github.com/sony/nnabla-ext-cuda
chainerからforkされたpytorchはライブラリ内でcudaファイルがたくさんあってそこで繋いでる。
海外のコミュニティ見てても、なんで出したの？why??
的なコメントが多数。
個人的には日本の企業なので応援したいので、時間があったら使っていこうかと思う。