nvidia-dockerを使ったchainerのインストールについて

[wakuwaku Ukiuki]

nvidia-dockerのコンテナにインストールしたchainer1.19でトレーニング用のスクリプトを実行させますと、下記のようなエラーが出ます。

/root/anaconda2/lib/python2.7/site-packages/chainer/cuda.py:88: UserWarning: cuDNN is not enabled.

Please reinstall chainer after you install cudnn

(see https://github.com/pfnet/chainer#installation).

  'cuDNN is not enabled.\n'

/root/anaconda2/lib/python2.7/site-packages/chainer/function_set.py:62: FutureWarning: 'collect_parameters' is deprecated. You can pass FunctionSet itself to 'optimizer.setup'

  warnings.warn(msg, FutureWarning)

質問：nvidia-dockerはcudnnがパッケージされていますので、chainerは後からインストールになりまが、cudnnを利用したい場合、nvidia-dockerを利用してビルドしないほうが良いですか？

よろしくお願いいたします。

ちなみに、chiner 1.22ではエラーは出ませんでした。

[wakuwaku Ukiuki]

すみません。訂正があります。失礼しました。

chainer 1.22.0でも同じエラーが出ていました。

/root/anaconda3/lib/python3.5/site-packages/chainer/cuda.py:89: UserWarning: cuDNN is not enabled.

[Yuya Unno]

どのbase imageを使いましたか？

Chainerはcudnnのヘッダファイルなどを使ってビルドを行うため，runtime版ではなくdevel版（例えば8.0-cudnn5-devel-ubuntu14.04）を使う必要があります．

https://hub.docker.com/r/nvidia/cuda/

もっと簡単には，公式イメージを使ってください．

https://hub.docker.com/r/chainer/chainer/

2017年3月29日 1:26 wakuwaku Ukiuki <ukiuki2...@gmail.com>:

--
このメールは Google グループのグループ「Chainer Japanese User Group」に登録しているユーザーに送られています。
このグループから退会し、グループからのメールの配信を停止するには chainer-jp+unsubscribe@googlegroups.com にメールを送信してください。
このグループに投稿するには chain...@googlegroups.com にメールを送信してください。
このディスカッションをウェブ上で閲覧するには https://groups.google.com/d/msgid/chainer-jp/b52e6b70-edb4-4758-a1b5-ed7af49b8d17%40googlegroups.com にアクセスしてください。
その他のオプションについては https://groups.google.com/d/optout にアクセスしてください。

[wakuwaku Ukiuki]

こんにちは。解決方法があってるか分かりませんがエラー（正確には警告ですが）が解決しました。

※ Yuya Unno Wrote:Chainerはcudnnのヘッダファイルなどを使ってビルドを行うため，runtime版ではなくdevel版（例えば8.0-cudnn5-devel-ubuntu14.04）を使う必要があります．

頂いたヒントよりnvidia-dockerにchainerをインストールする際、cudnnライブラリをビルドしていないと言うことだと理解しました。

やったこと：Dockerfileに下記記載の順に追加してビルド

1.cudnnインストールの記述追加

2.chainerインストールの記述追加

1. https://gitlab.com/nvidia/cuda/blob/ubuntu16.04/8.0/devel/cudnn5/Dockerfile

2. https://hub.docker.com/r/chainer/chainer/~/dockerfile/　

・Anacondaを使いたかったので記述を少し改造してwgetでインストール

・chainer1.16を使いたかったので記述を少し修正

出来上がったイメージ

Anaconda-python2.7

chainer1.16

その他、個人的に利用したいツール

このイメージを利用できる環境

ubuntu16.04

gtx1070 or gtx1080

ありがとうございました。