コンテンツにスキップするには Enter キーを押してください

ubuntu でcuDNN・chainerをインストールする。

はじめに

ubuntuで学習を回すために、3回に渡ってubuntuでグラボのドライバをインストールするまでを行いました。






 

cuDNNのインストール

すでに、以前のブログの状態でCUDAはインストールしてあります。
cuDNNとは、ディープラーニング用のCUDAのライブラリとありました。
これは、NVIDIAの サイト からダウンロードします。
ダウンロードするためにはログインする必要があり、無料ではありますが登録しなければなりません。
「Download cuDNN v5.1 (Jan 20, 2017), for CUDA 8.0」の「cuDNN v5.1 Library for Linux」をダウンロードします。
tgzという様式で圧縮されているので、tarコマンドでもGUI上でもいいので解凍しておきます。

次に、.bashrcや.zshrcに以下を記述します。

...
export CUDA_PATH=/usr/local/cuda-8.0
...

 
解凍したディレクトリに移動し、中のファイルを次の通りに移動させます。

~/cuda $ sudo cp include/* $CUDA_PATH/include
~/cuda $ sudo cp lib64/* $CUDA_PATH/lib64

 

pythonの設定

ubuntuには最初からpythonが入っていますが、いろいろなパッケージが最初から入っているanacondaをおすすめします。
pythonのバージョン管理にはpyenvを用います。ubuntuでのpyenvについては以前書きました。


 

~$ pyenv install anaconda3-4.2.0
~$ pyenv global anaconda3-4.2.0
~$ pyenv versions

anacondaに*がついていて、選択されていることを確認します。

 

chainer のインストール

cuDNNをCUDA_PATHが示す場所に正しくインストールしていると、chainerのインストール時にそこが参照されcuDNNを使用できるようになるそうです。

~$ pip install chainer
~$ python
> import chainer
> print(chainer.__version__)
1.21.0
> import cupy
> import cupy.cudnn

import をした後に、エラーが出なければ正しくインストール出来ています。

最後にサンプルを動かします。

~$ git clone https://github.com/pfnet/chainer
~$ cd chainer/examples/mnist
~$ python train_mnist.py -g 0

これで、学習が始まれば終わりです。

しかし、2017/03/07現在、githubからmasterだけどダウンロードしてそのまま動かしても動かないという記事がありました。
同様に筆者も動きませんでした。
エラーは以下のとおりです。

> $ python train_mnist.py                                                    [±master ✓]
GPU: -1
# unit: 1000
# Minibatch-size: 100
# epoch: 20

Traceback (most recent call last):
  File "train_mnist.py", line 125, in <module>
    main()
  File "train_mnist.py", line 96, in main
    if extensions.PlotReport.available():
AttributeError: type object 'PlotReport' has no attribute 'available'

あたらしいextensionsのPlotReportにavailableなんてないといわれます。
なので、これをコメントアウトすれば普通に動きますが、なんかスッキリしません。
ただ、chainer自体は問題なく過去のコードも動くので、この問題はいずれ解決するとして置いときましょう。

では〜





コメントする

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です