satzz / chainer

A flexible framework of neural networks for deep learning

chainer: Neural network framework

Requirements

Minimum requirements:

• Python 2.7+ (v2.7.9 is recommended)
• NumPy

Requirements to enable CUDA:

• CUDA 6.5+
• PyCUDA
• scikits.cuda (pip install 'scikits.cuda>=0.5.0b1,!=0.042')
• Mako (depending through PyCUDA)

Recommended:

• CuDNN v2
• scikit-learn (to run some examples)
• OpenCV 2.4 (to run some examples)

Installation

Install Chainer via PyPI:

pip install chainer

You can also install Chainer from the source code:

python setup.py install

If you want to enable CUDA, first you have to install CUDA and set the environment variable CUDA_DIR to the installed path. You also have to set binary and library paths by appropriate environment variables like PATH and LD_LIBRARY_PATH. Then install CUDA-related packages by:

pip install chainer-cuda-deps

or, from the source:

python cuda_deps/setup.py install

License

MIT License (see LICENSE file).