Chainer+CUDNN+CUDA

The reason I wanna use Chainer:

  1. I am confused the way of install Caffe
  2. I wanna use the available Fast-CNN code
    1. https://github.com/mitmul/chainer-faster-rcnn

Chainer is really easy to install, just follow the home-page

http://docs.chainer.org/en/stable/install.html

If wanna use gpu in the mnist example:

  1. mnist$ python train_mnist.py –gpu=0      GPU using
  2. mnist$ python train_mnist.py                       CPU using

Debug

  1. Try the sample can not from chainer import training
    solution: reinstal chainer
  2. To use Cudnn and CUDA
    1. Need cudnn.h
    2. libcudnn.so*

To check it use CUDA or not

>>> from chainer import cuda
>>> cuda.check_cuda_available()

I hope to have a deep understanding of Chainer in the near future!

Advertisements

Leave a Reply

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out / Change )

Twitter picture

You are commenting using your Twitter account. Log Out / Change )

Facebook photo

You are commenting using your Facebook account. Log Out / Change )

Google+ photo

You are commenting using your Google+ account. Log Out / Change )

Connecting to %s