CUDA8.0とcuDNN6のubuntuへのインストール

既にCUDAがインストールされているイメージを使ってdockerのコンテナを作ったのだが,ドキュメントがなかったので動作させることができなかった.結局,debパッケージを拾ってきてインストールした.ココがとても参考になった.

その後,chainerをpipでインストールしたのだが,

python
>>> import cupy
>>> import cupy.cudnn

でエラーとなる.???と思ったが,cupyは自分でpipでインストールしなければならなかっただけだ.cupyをインストールした後はcupyもcupy.cudnnもimportできた.

広告