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

pandazx's blog

Hadoop, データ分析など雑多な技術ブログ

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