EC2のGPUインスタンスにChainerインストール
前提
基本的には以下のサイトを参考にすればよい。
その際に、ハマったことだけメモ。
最初、NVIDIA GPU DRIVER入りのAMIを使って、EC2を作成したが、CUDA ver6.5であることに気づかず、cuDNN v4をインストールしてしまった。 この状態で、サンプルのtrain_mnist.py を実行すると以下のWARNINGが出る。
'cuDNN is not enabled'
CUDA ver6.5 に対応するcuDNN v2をインストールして、Chainerを再インストールしたが、状況は変わらなかった。公式ドキュメントに書いてあるが、 以下のように、クリーンインストールすることで解決。
pip uninstall chainer pip install chainer --no-cache-dir
Chainer公式ドキュメント:Install Guide — Chainer 1.7.2 documentation