TadaoYamaokaの日記

山岡忠夫Homeで公開しているプログラムの開発ネタを中心に書いていきます。

WindowsにChainer v3+CUDA9+cuDNN7をインストールする

Chainer v3(cupy v2)がCUDA9に対応したので、バージョンアップしました。
Chainer v3はcuDNNも最新のバージョン7に対応しているので、cuDNNも7にしました。

インストール手順は、以前のバージョンと同様です。

chainerをバージョンアップする際は、chainerとcupyをアンインストールしてからインストールします。

pip uninstall chainer
pip uninstall cupy
pip install cupy --no-cache-dir
pip install chainer --no-cache-dir

性能比較

CUDA8+cuDNN5.1と学習速度を比較しました。

比較対象として、将棋AIのResNet(5ブロック)の1000万局面の学習時間で比較しました。

CUDA8+cuDNN5.1 1:09:18
CUDA9+cuDNN7 1:07:16

※2回測定の平均
GPUは、GeForce 1080Ti

約3%学習速度が速くなりました。

なお、速度向上の要因がCUDA9によるものか、cuDNN6によるものか、cuDNN7によるものかは分かりません。