windowsで開発環境を整える(cudnn導入)

cudnnを入れます。

cudnnのインストール

Please reinstall chainer after you install cudnn

と出ていたので、cudnnを入れたいです。

NVIDIA cuDNN – GPU Accelerated Deep Learning | NVIDIA Developer
からダウンロードしたいが、その前に申請が必要らしく、↑のページのRegisterから登録申請。
数日待ってメールが来たら、ダウンロードできるようになるようです。

ページを進めて、「cuDNN v3 Library for Windows」からzipファイルをダウンロード。
(面倒なので、)解凍したファイルをインストールした「NVIDIA GPU Computing Toolkit/」以下に手動でコピーしてしまいます。

これで終了。

chainerのリビルド

anaconda promptを開いて、

$ activate py34
$ cd chainerを置いてる場所
$ python setup.py install

確認

$ python examples\mnist\train_mnist.py --gpu=0
load MNIST dataset
epoch 1
kern.cu
graph generated
train mean loss=0.1948567812796682, accuracy=0.9418333357013762
test  mean loss=0.09421748955908697, accuracy=0.9709000051021576
...
epoch 20
train mean loss=0.006696289535010464, accuracy=0.998050001859665
test  mean loss=0.11188006146020016, accuracy=0.9816000056266785
save the model
save the optimizer

エラーなく実行できました。