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
エラーなく実行できました。