How to install cuDNN

cuDNN provides primitives for deep learning networks that have been accelerated for GPUs by NVIDIA.

  • To download cuDNN head over to the cuDNN page here. cuDNN is not directly available for download. NVIDIA requires you to create a login. After that it presents cuDNN downloads in different formats (.tgz or .deb).

  • I prefer to install from .tgz since it gives more control. Unzip the file and it will create a cuda directory which has the required include and lib directories.

  • Copy the files over to your CUDA directory:

$ sudo cp cuda/include/cudnn.h /usr/local/cuda-X.Y/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-X.Y/lib64
$ sudo chmod a+r /usr/local/cuda-X.Y/include/cudnn.h
$ sudo chmod a+r /usr/local/cuda-X.Y/lib64/libcudnn*

Replace cuda-X.Y with the specific CUDA version directory on your system.

That is it! Caffe and other projects needing cuDNN should be able to find and link with cuDNN now.

Tried with: cuDNN 6.0 and Ubuntu 16.04

3 thoughts on “How to install cuDNN

Leave a Reply

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

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

Google photo

You are commenting using your Google 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 )

Connecting to %s

This site uses Akismet to reduce spam. Learn how your comment data is processed.