前提
- Python 2.7.6+, 3.4.3+, 3.5.1+, 3.6.0+が入っている
- NumPy 1.9, 1.10, 1.11, 1.12が使える
- six 1.9が使える
- cythonが入っている
- virtualenvが使える
CUDA
- CUDA 8.0のインストールが済んでいる
- g++ 4.8.4+が入っている
Install
手順
- 適当な場所で
git clone git@github.com:pfnet/chainer.git
- virtualenvの環境作成
~/dev/chainer$ virtualenv venv --python=/usr/bin/python3.4
~/dev/chainer$ source activate venv/bin/activate
- cuDNN用の環境マネージメントツールcudnnenvでcuDNNを入れる
$ pip install cudnnenv
$ cudnnenv install v5-cuda8
$ cudnnenv activate v5-cuda8
$ cudnnenv version
v5-cuda8
- 必要な環境変数を設定する
$ tail -n 11 ~/.bashrc
#path for cuDNN
export LD_LIBRARY_PATH=~/.cudnn/active/cuda/lib64:$LD_LIBRARY_PATH
export CPATH=~/.cudnn/active/cuda/include:$CPATH
export LIBRARY_PATH=~/.cudnn/active/cuda/lib64:$LIBRARY_PATH
# path for chainer
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
export CFLAGS=-I~/.cudnn/active/cuda/include
export LDFLAGS=-L~/.cudnn/active/cuda/lib64
export LD_LIBRARY_PATH=~/.cudnn/active/cuda/lib64:$LD_LIBRARY_PATH
- 開発者ようにchainerをビルドする
~/dev/chainer$ python setup.py develop
...
...
Finished processing dependencies for chainer==1.21.0