読者です 読者をやめる 読者になる 読者になる

Deep Learningのフレームワーク近況

慣れないPythonチュートリアルを勉強中。
課題でXOR分類器をコーディングしてみたり。

本題。

tensorflow
0.7.1がリリース。
Download and Setupの手順も更新されている。(こういうところ、すごく親切)

linux GPU版だけ引用する。

# Ubuntu/Linux 64-bit, GPU enabled:
$ sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.7.1-cp27-none-linux_x86_64.whl

CUDA toolkit 7.5 と cuDNNのv4対応版のバイナリと思われる。

CNTK
特に動きはなさそう。
ImageReaderに手が入ったっぽい。

chainer
1.7.0がリリース。
こちらもcuDNN v4に対応したようだ。

整理すると、
・CUDA toolkit 7.5 + cuDNN v4
tensorflow 0.7.1
chainer 1.7.0
→ どちらも問題なくGPUで動作した。

・CUDA toolkit 7.0 + cuDNN v4
CNTK beta(2016.02.08)

以上。