Skip to content

Latest commit

 

History

History
61 lines (48 loc) · 1.88 KB

README.md

File metadata and controls

61 lines (48 loc) · 1.88 KB

pypi pypi GitHub license travis coveralls

Chainer: a neural network framework

Requirements

Minimum requirements:

  • Python 2.7+, 3.4+ or 3.5+
  • NumPy 1.9+
  • Six 1.9+

Requirements for some features:

  • CUDA support
    • CUDA 6.5+
    • filelock
  • CuDNN support
    • CuDNN v2
  • Caffe model support
    • Python 2.7+ (Py3 is not supported)
    • Protocol Buffers (pip install protobuf)
  • Testing utilities
    • Mock
    • Nose

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 PATH and LD_LIBRARY_PATH for CUDA executables and libraries. For example, if you are using Ubuntu and CUDA is installed by the official distribution, then CUDA is installed at /usr/local/cuda. In this case, you have to add the following line to .bashrc or .zshrc (choose which you are using):

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

Do not forget to restart your terminal session (or source it) to enable this change. It automatically enables the CUDA support.

More information

License

MIT License (see LICENSE file).