Difference between revisions of "Jetson/cuDNN"

From eLinux.org
Jump to: navigation, search
(Created a cuDNN page)
 
(Added version information and link to a github gist with the commands to run.)
 
Line 1: Line 1:
cuDNN is NVIDIA's GPU-accelerated library of primitives for deep neural networks and machine learning. It is integrated into higher-level machine learning frameworks such as UC Berkeley’s [http://caffe.berkeleyvision.org Caffe] software, and is now available for [[Jetson TK1]].
+
cuDNN is NVIDIA's GPU-accelerated library of primitives for deep neural networks and machine learning [https://developer.nvidia.com/cuDNN]. It is integrated into higher-level machine learning frameworks such as UC Berkeley’s [http://caffe.berkeleyvision.org Caffe] software, and is now available for [[Jetson TK1]].  cuDNN Release 1 is the current stable release, and cuDNN v2 is at release candidate 3.  At this time, Caffe uses cuDNN release 1.  All libraries and documentation are available from the NVIDIA website, which requires a no-cost developer registration to download.
  
There is an overview of cuDNN for embedded on the [http://devblogs.nvidia.com/parallelforall/embedded-machine-learning-cudnn-deep-neural-network-library-jetson-tk1/ Parallel ForAll Blog], and you can download it from its [https://developer.nvidia.com/cuDNN webpage].
+
An overview of cuDNN for embedded is on the [http://devblogs.nvidia.com/parallelforall/embedded-machine-learning-cudnn-deep-neural-network-library-jetson-tk1/ Parallel ForAll Blog].
 +
 
 +
A step-by-step procedure to install cuDNN on the Jetson is available as a GitHub gist at [https://gist.github.com/jetsonhacks/fa9f4ff89006607359ea].  After installing the cuDNN library, the Caffe <code>Makefile.config</code> must be modified to use cuDNN.
 +
 
 +
    # Makefile.config (only relevant cuDNN configuration items shown)
 +
    USE_CUDNN := 1
 +
    INCLUDE_DIRS := /usr/local/cuda-6.5/include
 +
    LIBRARY_DIRS := /usr/local/cuda-6.5/lib
 +
 
 +
After these configuration changes, the Caffe source code must be recompiled.
 +
 
 +
The NVIDIA documentation contains information on how to develop software code that utilizes cuDNN (caveat: the version 2 rc3 documentation is more thorough).

Latest revision as of 09:10, 8 March 2015

cuDNN is NVIDIA's GPU-accelerated library of primitives for deep neural networks and machine learning [1]. It is integrated into higher-level machine learning frameworks such as UC Berkeley’s Caffe software, and is now available for Jetson TK1. cuDNN Release 1 is the current stable release, and cuDNN v2 is at release candidate 3. At this time, Caffe uses cuDNN release 1. All libraries and documentation are available from the NVIDIA website, which requires a no-cost developer registration to download.

An overview of cuDNN for embedded is on the Parallel ForAll Blog.

A step-by-step procedure to install cuDNN on the Jetson is available as a GitHub gist at [2]. After installing the cuDNN library, the Caffe Makefile.config must be modified to use cuDNN.

   # Makefile.config (only relevant cuDNN configuration items shown)
   USE_CUDNN := 1
   INCLUDE_DIRS := /usr/local/cuda-6.5/include
   LIBRARY_DIRS := /usr/local/cuda-6.5/lib

After these configuration changes, the Caffe source code must be recompiled.

The NVIDIA documentation contains information on how to develop software code that utilizes cuDNN (caveat: the version 2 rc3 documentation is more thorough).