NVCaffe Overview

NVCaffe Overview (PDF)

The NVIDIA Deep Learning SDK accelerates widely-used deep learning frameworks such as NVCaffe™ .

Caffe™ was originally developed by the Berkeley Vision and Learning Center (BVLC) and by community contributors. It is a deep learning framework made with expression, speed, and modularity in mind.

NVCaffe is an NVIDIA-maintained fork of BVLC Caffe tuned for NVIDIA GPUs, particularly in multi-GPU configurations, accelerated by the NVIDIA Deep Learning SDK. It includes multi-precision support as well as other NVIDIA-enhanced features and offers performance specially tuned for NVIDIA GPU systems.

The NVCaffe framework is pre-built, tested, ready to run, and includes all of the necessary dependencies. The NVCaffe source code is also included for those who want to augment it with custom modifications or enhancements.

See /workspace/README.md inside the container for information on customizing your NVCaffe image. For more information about Caffe, including tutorials, documentation, and examples, see the Caffe website.

NVCaffe typically utilizes the same input formats and configuration parameters as Caffe, therefore, community-authored materials and pre-trained models for Caffe can usually also be applied to NVCaffe.

This document describes the key features, software enhancements and improvements, any known issues, and how to run this container.